From 73d3d439b159cc1fc7a8556e380b102100a5d7c7 Mon Sep 17 00:00:00 2001 From: Scott Lindauer Date: Wed, 31 Jan 2024 11:08:39 -0500 Subject: [PATCH] Black reformatting (second attempt) --- src/sasctl/_services/model_repository.py | 7 +++---- src/sasctl/pzmm/write_json_files.py | 6 +++--- 2 files changed, 6 insertions(+), 7 deletions(-) diff --git a/src/sasctl/_services/model_repository.py b/src/sasctl/_services/model_repository.py index 37a4534f..0971a125 100644 --- 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 index 2037aac9..4972dab2 100644 --- 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,