Adjust test

duydl · duydl · commit f2b082b93c35 · 2025-02-21T20:58:29.000+07:00
diff --git a/tests/tests_pytorch/loggers/test_csv.py b/tests/tests_pytorch/loggers/test_csv.py
@@ -116,7 +116,6 @@ def test_log_hyperparams(tmp_path):
         "layer": torch.nn.BatchNorm1d,
     }
     logger.log_hyperparams(hparams)
-    logger.save()
 
     path_yaml = os.path.join(logger.log_dir, ExperimentWriter.NAME_HPARAMS_FILE)
     params = load_hparams_from_yaml(path_yaml)

Original file line number	Diff line number	Diff line change
`@@ -116,7 +116,6 @@ def test_log_hyperparams(tmp_path):`
`116`	`116`	`"layer": torch.nn.BatchNorm1d,`
`117`	`117`	`}`
`118`	`118`	`logger.log_hyperparams(hparams)`
`119`		`- logger.save()`
`120`	`119`
`121`	`120`	`path_yaml = os.path.join(logger.log_dir, ExperimentWriter.NAME_HPARAMS_FILE)`
`122`	`121`	`params = load_hparams_from_yaml(path_yaml)`