@@ -141,14 +141,13 @@ def update_kpis(
141
141
models_to_update = kpis ["ModelUUID" ].unique ().tolist ()
142
142
143
143
for model in models_to_update :
144
- try :
145
- current_params , file_name = _find_file (model , "hyperparameters" )
146
- except :
147
- model_name = {kpis .loc [kpis ["ModelUUID" ]== model , "ModelName" ].iloc [0 ]}
148
- print (f'No hyperparameter file for current model { model_name } . Attempting for next model...' )
149
- else :
150
- updated_json = cls ._update_json (model , current_params .json (), kpis )
151
- mr .add_model_content (model , json .dumps (updated_json , indent = 4 ), file_name )
144
+ try :
145
+ current_params , file_name = _find_file (model , "hyperparameters" )
146
+ except :
147
+ print (f'No hyperparamter file for current model { kpis .loc [kpis ["ModelUUID" ]== model , "ModelName" ].iloc [0 ]} . Attempting for next model...' )
148
+ else :
149
+ updated_json = cls ._update_json (model , current_params , kpis )
150
+ mr .add_model_content (model , json .dumps (updated_json , indent = 4 ), file_name )
152
151
153
152
@staticmethod
154
153
def get_hyperparameters (model : Union [str , dict , RestObj ]) -> Tuple [dict , str ]:
@@ -175,7 +174,7 @@ def get_hyperparameters(model: Union[str, dict, RestObj]) -> Tuple[dict, str]:
175
174
model = mr .get_model (model )
176
175
id_ = model ["id" ]
177
176
file_contents , file_name = _find_file (id_ , "hyperparameters" )
178
- return file_contents . hyperparameters , file_name
177
+ return file_contents , file_name
179
178
180
179
@classmethod
181
180
def add_hyperparameters (cls , model : Union [str , dict , RestObj ], ** kwargs ) -> None :
@@ -201,7 +200,7 @@ def add_hyperparameters(cls, model: Union[str, dict, RestObj], **kwargs) -> None
201
200
id_ = model ["id" ]
202
201
hyperparameters , file_name = cls .get_hyperparameters (id_ )
203
202
for key , value in kwargs .items ():
204
- hyperparameters [key ] = value
203
+ hyperparameters ["hyperparameters" ][ key ] = value
205
204
mr .add_model_content (
206
205
model ,
207
206
json .dumps (hyperparameters , indent = 4 ),
0 commit comments