Skip to content

Commit 8599b27

Browse files
committed
Fixed issue with mappingproxy __dict__ in train #268
1 parent 2db7aec commit 8599b27

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

comet/cli/train.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)