chore: fix pre-commit

Author: Timon Viola

src/dagcellent/operators/mlflow/__init__.py

@@ -8,6 +8,7 @@
     Model registration, search, artifact loading, and deployment
     Execution of GitHub projects within notebook & against a remote target.
 """
+
 from __future__ import annotations
 
 from dagcellent.operators.mlflow._operators import (
@@ -16,10 +17,9 @@
     GetModelVersionByNameAndStage,
     SetModelVersionTag,
 )
-
 from dagcellent.operators.mlflow._utils import (
-    SlimModelVersion,
     MlflowModelStage,
+    SlimModelVersion,
 )
 
 __all__ = [

src/dagcellent/operators/mlflow/_operators.py

@@ -75,8 +75,8 @@ class GetModelVersionByNameAndStage(BaseOperator):
         ```python
         # Get prod model run id
         from dagcellent.operators.mlflow import (
-          GetModelVersionByNameAndStage,
-          MlflowModelStage,
+            GetModelVersionByNameAndStage,
+            MlflowModelStage,
         )
 
         get_prod_run_id = GetModelVersionByNameAndStage(

tests/test_mlflow.py

@@ -5,4 +5,5 @@
 
 def test_import():
     from dagcellent.operators.mlflow import GetModelMetaData
+
     assert GetModelMetaData is not None