better art typing

simplymathematics · Aug 13, 2024 · 658e987 · 658e987
1 parent db2c9ed
commit 658e987
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/deckard/base/model/sklearn_pipeline.py b/deckard/base/model/sklearn_pipeline.py
@@ -179,7 +179,9 @@ def __call__(self, model):
                     Pipeline,
                 ):
                     model.model = Pipeline([("model", model.model)])
-                elif str(type(model)).startswith("art.") and isinstance(model.model, Pipeline):
+                elif str(type(model)).startswith("art.") and isinstance(
+                    model.model, Pipeline
+                ):
                     model.model.steps.insert(-2, [stage, model.model])
                 else:
                     model.steps.insert(-2, [stage, model])
@@ -256,4 +258,4 @@ def __call__(self):
             model,
             "fit",
         ), f"model must have a fit method. Got type {type(model)}"
-        return model
+        return model