Attempt lint fix

nextmv-io · Nov 22, 2024 · a0d1267 · a0d1267
1 parent fdf6bd8
commit a0d1267
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nextmv/model.py b/nextmv/model.py
@@ -122,7 +122,7 @@ def save(self, model_dir: str, configuration: ModelConfiguration) -> None:
         """
 
         try:
-            import mlflow
+            import mlflow as mlflow
         except ImportError:
             # If mlflow is not installed, we install it.
             import subprocess