File tree Expand file tree Collapse file tree 1 file changed +8
-3
lines changed
sdk/python/v1beta1/kubeflow/katib/utils Expand file tree Collapse file tree 1 file changed +8
-3
lines changed Original file line number Diff line number Diff line change @@ -188,8 +188,9 @@ def get_trial_substitutions_from_trainer(
188
188
if isinstance (parameters , TrainingArguments ):
189
189
parameters_dict = parameters .to_dict ()
190
190
else :
191
- parameters_dict = parameters .__dict__
192
-
191
+ parameters_dict = (
192
+ parameters .to_dict () if hasattr (parameters , "to_dict" ) else vars (parameters )
193
+ )
193
194
for p_name , p_value in parameters_dict .items ():
194
195
if not hasattr (parameters , p_name ):
195
196
logger .warning (f"Training parameter { p_name } is not supported." )
@@ -218,7 +219,11 @@ def get_trial_substitutions_from_trainer(
218
219
if isinstance (parameters , TrainingArguments ):
219
220
parameters = json .dumps (parameters .to_dict ())
220
221
else :
221
- parameters = json .dumps (parameters .__dict__ , cls = SetEncoder )
222
+ parameters = (
223
+ json .dumps (parameters .to_dict (), cls = SetEncoder )
224
+ if hasattr (parameters , "to_dict" )
225
+ else json .dumps (vars (parameters ), cls = SetEncoder )
226
+ )
222
227
223
228
return parameters
224
229
You can’t perform that action at this time.
0 commit comments