Skip to content

Commit 9646d98

Browse files
committed
changes to ensure all integration tests pass
1 parent 9d839b8 commit 9646d98

File tree

1 file changed

+9
-11
lines changed

1 file changed

+9
-11
lines changed

src/sasctl/pzmm/write_json_files.py

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1300,17 +1300,15 @@ def calculate_model_statistics(
13001300
roc_dict = cls.apply_dataframe_to_json(json_dict[1]["data"], i, roc_df)
13011301
for j in range(len(roc_dict)):
13021302
json_dict[1]["data"][j].update(roc_dict[j])
1303-
if(roc_dict[j]["dataMap"]["_KS_"] == 1):
1304-
fitstat_data = {
1305-
"_KS_": roc_dict[j]["dataMap"]["_KS_"],
1306-
"_KS2_": roc_dict[j]["dataMap"]["_KS2_"],
1307-
"_C_": roc_dict[j]["dataMap"]["_C_"],
1308-
"_Gini_": roc_dict[j]["dataMap"]["_Gini_"],
1309-
"_Gamma_": roc_dict[j]["dataMap"]["_Gamma_"],
1310-
"_Tau_": roc_dict[j]["dataMap"]["_Tau_"]
1311-
}
1312-
1313-
json_dict[0]["data"][i]["dataMap"].update(fitstat_data)
1303+
fitstat_data = None
1304+
if roc_dict[j]["dataMap"]["_KS_"] == 1:
1305+
fitstat_data = dict()
1306+
missing_stats = "_KS_", "_KS2_", "_C_", "_Gini_", "_Gamma_", "_Tau_"
1307+
for stat in missing_stats:
1308+
if stat in roc_dict[j]["dataMap"]:
1309+
fitstat_data[stat] = roc_dict[j]["dataMap"][stat]
1310+
if fitstat_data:
1311+
json_dict[0]["data"][i]["dataMap"].update(fitstat_data)
13141312

13151313
lift_df = pd.DataFrame(conn.CASTable("Lift", caslib="Public").to_frame())
13161314
lift_dict = cls.apply_dataframe_to_json(json_dict[2]["data"], i, lift_df, 1)

0 commit comments

Comments
 (0)