black reformatting

Author: djm21

src/sasctl/pzmm/write_score_code.py

@@ -250,7 +250,7 @@ def score(var1, var2, var3, var4):
                 input_var_list,
                 missing_values=missing_values,
                 dtype_list=input_dtypes_list,
-                preprocess_function=preprocess_function
+                preprocess_function=preprocess_function,
             )
             self._predictions_to_metrics(
                 score_metrics,
@@ -267,7 +267,7 @@ def score(var1, var2, var3, var4):
                 missing_values=missing_values,
                 statsmodels_model="statsmodels_model" in kwargs,
                 tf_model="tf_keras_model" in kwargs or "tf_core_model" in kwargs,
-                preprocess_function=preprocess_function
+                preprocess_function=preprocess_function,
             )
             # Include check for numpy values and a conversion operation as needed
             self.score_code += (
@@ -2263,7 +2263,9 @@ def _viya35_score_code_import(
                     mr.update_model(model)
                     return mas_code, cas_code
 
-    def _add_preprocess_code(self, preprocess_function: Callable[[DataFrame], DataFrame]):
+    def _add_preprocess_code(
+        self, preprocess_function: Callable[[DataFrame], DataFrame]
+    ):
         """
         Places the given preprocess function, which must both take a DataFrame as an argument
         and return a DataFrame, into the score code. If the preprocess function does not

tests/unit/test_write_score_code.py

@@ -188,6 +188,7 @@ def test_impute_missing_values():
     assert "'b': 'test'" in sc.score_code
     assert "'c': 1" in sc.score_code or "'c': np.int64(1)" in sc.score_code
 
+
 def test_preprocess_function():
     """
     Test Cases:
@@ -199,20 +200,24 @@ def test_preprocess_function():
     )
     sc = ScoreCode()
     sc.score_code = " "
+
     def preprocess_function_one(data: pd.DataFrame):
         print("preprocessing happens here")
         return data
+
     sc._add_preprocess_code(preprocess_function_one)
     assert "preprocessing happens here" in sc.score_code
     assert "preprocess_function_one" in sc.score_code
 
     sc = ScoreCode()
     sc.score_code = " "
+
     def preprocess_function_two(data: pd.DataFrame):
         print("preprocessing happens here?")
+
     with pytest.raises(ValueError):
         sc._add_preprocess_code(preprocess_function_two)
-    
+
 
 def test_predict_method():
     """