Skip to content

Commit c680342

Browse files
committed
fixed issue with find_flie in model_parameters
1 parent 841d8ec commit c680342

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sasctl/pzmm/model_parameters.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def update_kpis(
146146
except:
147147
print(f'No hyperparamter file for current model {kpis.loc[kpis["ModelUUID"]==model, "ModelName"].iloc[0]}. Attempting for next model...')
148148
else:
149-
updated_json = cls._update_json(model, current_params.json(), kpis)
149+
updated_json = cls._update_json(model, current_params, kpis)
150150
mr.add_model_content(model, json.dumps(updated_json, indent=4), file_name)
151151

152152
@staticmethod
@@ -174,7 +174,7 @@ def get_hyperparameters(model: Union[str, dict, RestObj]) -> Tuple[dict, str]:
174174
model = mr.get_model(model)
175175
id_ = model["id"]
176176
file_contents, file_name = _find_file(id_, "hyperparameters")
177-
return file_contents.json(), file_name
177+
return file_contents, file_name
178178

179179
@classmethod
180180
def add_hyperparameters(cls, model: Union[str, dict, RestObj], **kwargs) -> None:

0 commit comments

Comments
 (0)