@@ -89,7 +89,7 @@ def initialize_trainer(configs) -> Trainer:
8989 lr_monitor = LearningRateMonitor (logging_interval = "step" )
9090 trainer_args ["callbacks" ] = [early_stop_callback , checkpoint_callback , lr_monitor ]
9191 print ("TRAINER ARGUMENTS: " )
92- print (json .dumps (trainer_args , indent = 4 , default = lambda x : x . __dict__ ))
92+ print (json .dumps (trainer_args , indent = 4 , default = str ))
9393 trainer = Trainer (** trainer_args )
9494 return trainer
9595
@@ -101,7 +101,7 @@ def initialize_model(configs):
101101 json .dumps (
102102 configs .regression_metric .init_args ,
103103 indent = 4 ,
104- default = lambda x : x . __dict__ ,
104+ default = str ,
105105 )
106106 )
107107 if configs .load_from_checkpoint is not None :
@@ -120,7 +120,7 @@ def initialize_model(configs):
120120 json .dumps (
121121 configs .referenceless_regression_metric .init_args ,
122122 indent = 4 ,
123- default = lambda x : x . __dict__ ,
123+ default = str ,
124124 )
125125 )
126126 if configs .load_from_checkpoint is not None :
@@ -137,7 +137,7 @@ def initialize_model(configs):
137137 elif configs .ranking_metric is not None :
138138 print (
139139 json .dumps (
140- configs .ranking_metric .init_args , indent = 4 , default = lambda x : x . __dict__
140+ configs .ranking_metric .init_args , indent = 4 , default = str
141141 )
142142 )
143143 if configs .load_from_checkpoint is not None :
@@ -152,7 +152,7 @@ def initialize_model(configs):
152152 elif configs .unified_metric is not None :
153153 print (
154154 json .dumps (
155- configs .unified_metric .init_args , indent = 4 , default = lambda x : x . __dict__
155+ configs .unified_metric .init_args , indent = 4 , default = str
156156 )
157157 )
158158 if configs .load_from_checkpoint is not None :
0 commit comments