Black reformatting (second attempt)

sassoftware · Jan 31, 2024 · 73d3d43 · 73d3d43
1 parent 426282b
commit 73d3d43
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/src/sasctl/_services/model_repository.py b/src/sasctl/_services/model_repository.py
@@ -359,10 +359,9 @@ def create_model(
 
         model["scoreCodeType"] = score_code_type or model.get("scoreCodeType")
         model["trainTable"] = training_table or model.get("trainTable")
-        model[
-            "classificationEventProbabilityVariableName"
-        ] = event_prob_variable or model.get(
-            "classificationEventProbabilityVariableName"
+        model["classificationEventProbabilityVariableName"] = (
+            event_prob_variable
+            or model.get("classificationEventProbabilityVariableName")
         )
         model["classificationTargetEventValue"] = event_target_value or model.get(
             "classificationTargetEventValue"

diff --git a/src/sasctl/pzmm/write_json_files.py b/src/sasctl/pzmm/write_json_files.py
@@ -883,9 +883,9 @@ def assess_model_bias(
             # run assessBias, if levels=None then assessBias treats the input like a regression problem
             tables = conn.fairaitools.assessbias(
                 modelTableType="None",
-                predictedVariables=pred_values
-                if pred_values is not None
-                else prob_values,
+                predictedVariables=(
+                    pred_values if pred_values is not None else prob_values
+                ),
                 response=actual_values,
                 responseLevels=levels,
                 sensitiveVariable=x,