chore: fix pre-commit

dfds-data · Jan 17, 2025 · 7c57a03 · 7c57a03
1 parent 31934cc
commit 7c57a03
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/src/dagcellent/operators/mlflow/__init__.py b/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__ = [

diff --git a/src/dagcellent/operators/mlflow/_operators.py b/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(

diff --git a/tests/test_mlflow.py b/tests/test_mlflow.py
@@ -5,4 +5,5 @@
 
 def test_import():
     from dagcellent.operators.mlflow import GetModelMetaData
+
     assert GetModelMetaData is not None
Original file line number	Diff line number	Diff line change
Expand Up		@@ -5,4 +5,5 @@

		def test_import():
		from dagcellent.operators.mlflow import GetModelMetaData

		assert GetModelMetaData is not None