Black reformatting

Author: smlindauer

src/sasctl/pzmm/pickle_model.py

@@ -112,14 +112,12 @@ def pickle_trained_model(
                     model=trained_model,
                     force=True,
                     path=pickle_path,
-                    filename=f"{model_prefix}.pickle"
+                    filename=f"{model_prefix}.pickle",
                 )
             # For MOJO H2O models, gzip the model file and adjust the file extension
             elif is_h2o_model and pickle_path:
                 trained_model.save_mojo(
-                    force=True,
-                    path=pickle_path,
-                    filename=f"{model_prefix}.mojo"
+                    force=True, path=pickle_path, filename=f"{model_prefix}.mojo"
                 )
             elif is_binary_model or is_h2o_model:
                 raise ValueError(

tests/unit/test_pickle_model.py

@@ -73,9 +73,7 @@ def test_pickle_trained_model_h2o():
     x.remove(y)
 
     model = H2OGeneralizedLinearEstimator(
-        family="binomial",
-        model_id="test_model",
-        lambda_search=True
+        family="binomial", model_id="test_model", lambda_search=True
     )
     model.train(x=x, y=y, training_frame=data)
 

tests/unit/test_write_json_files.py

@@ -272,7 +272,9 @@ def test_write_file_metadata_json():
         with patch.object(jf, "notebook_output", True):
             capture_output = io.StringIO()
             sys.stdout = capture_output
-            jf.write_file_metadata_json(model_prefix="Test_Model", json_path=Path(tmp_dir))
+            jf.write_file_metadata_json(
+                model_prefix="Test_Model", json_path=Path(tmp_dir)
+            )
             assert (Path(tmp_dir) / "fileMetadata.json").exists()
             sys.stdout = sys.__stdout__
             assert "was successfully written and saved to " in capture_output.getvalue()
@@ -636,4 +638,4 @@ def test_create_requirements_json(change_dir):
     unittest.TestCase.maxDiff = None
     unittest.TestCase().assertCountEqual(
         json.loads(json_dict["requirements.json"]), expected
-    )
+    )