Black reformatting

Author: smlindauer

src/sasctl/_services/model_management.py

@@ -199,9 +199,7 @@ def create_performance_definition(
         # If no models were specified, search the supplied project for all models
         elif not models[0]:
             project = mr.get_project(project)
-            models = mr.list_models(
-                filter=f"eq(projectName, '{project.name}')"
-            )
+            models = mr.list_models(filter=f"eq(projectName, '{project.name}')")
         else:
             for i, model in enumerate(models):
                 models[i] = mr.get_model(model)

src/sasctl/pzmm/model_parameters.py

@@ -141,13 +141,17 @@ def update_kpis(
         models_to_update = kpis["ModelUUID"].unique().tolist()
 
         for model in models_to_update:
-           try:
-               current_params, file_name = _find_file(model, "hyperparameters")
-           except:
-               print(f'No hyperparameter file for current model {kpis.loc[kpis["ModelUUID"]==model, "ModelName"].iloc[0]}. Attempting for next model...')
-           else:
-               updated_json = cls._update_json(model, current_params, kpis)
-               mr.add_model_content(model, json.dumps(updated_json, indent=4), file_name)
+            try:
+                current_params, file_name = _find_file(model, "hyperparameters")
+            except:
+                print(
+                    f'No hyperparameter file for current model {kpis.loc[kpis["ModelUUID"]==model, "ModelName"].iloc[0]}. Attempting for next model...'
+                )
+            else:
+                updated_json = cls._update_json(model, current_params, kpis)
+                mr.add_model_content(
+                    model, json.dumps(updated_json, indent=4), file_name
+                )
 
     @staticmethod
     def get_hyperparameters(model: Union[str, dict, RestObj]) -> Tuple[dict, str]:

tests/integration/test_tasks.py

@@ -196,7 +196,11 @@ def test_register_model(self, sklearn_linear_model):
 
         # Register model and ensure attributes are set correctly
         model = register_model(
-            sk_model, self.MODEL_NAME, project=self.PROJECT_NAME, input_data=X, force=True
+            sk_model,
+            self.MODEL_NAME,
+            project=self.PROJECT_NAME,
+            input_data=X,
+            force=True,
         )
 
         assert isinstance(model, RestObj)