Skip to content

Commit 512d763

Browse files
committed
fix unit test mock issue
1 parent 467378d commit 512d763

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/tests_pytorch/loggers/test_all.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,8 @@ def log_metrics(self, metrics, step):
108108
if logger_class == CometLogger:
109109
logger.experiment.id = "foo"
110110
logger._comet_config.offline_directory = None
111-
logger.experiment.project_name = "bar"
111+
logger._project_name = "bar"
112+
logger.experiment.get_key.return_value = "SOME_KEY"
112113

113114
if logger_class == NeptuneLogger:
114115
logger._retrieve_run_data = Mock()

0 commit comments

Comments
 (0)