From 4af64964b8a58a520164307f3c458d25cf0d0f75 Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Wed, 5 Aug 2020 21:29:03 +0200 Subject: [PATCH] tests --- tests/loggers/test_csv.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/loggers/test_csv.py b/tests/loggers/test_csv.py index e4de63e6185927..f2e8ab85fb9e56 100644 --- a/tests/loggers/test_csv.py +++ b/tests/loggers/test_csv.py @@ -4,7 +4,9 @@ import torch import os +from pytorch_lightning.core.saving import load_hparams_from_yaml from pytorch_lightning.loggers import CSVLogger +from pytorch_lightning.loggers.csv_logs import ExperimentWriter def test_file_logger_automatic_versioning(tmpdir): @@ -68,6 +70,10 @@ def test_file_logger_log_metrics(tmpdir, step_idx): logger.log_metrics(metrics, step_idx) logger.save() + path_yaml = os.path.join(logger.log_dir, ExperimentWriter.NAME_HPARAMS_FILE) + params = load_hparams_from_yaml(path_yaml) + assert all([n in params for n in metrics]) + def test_file_logger_log_hyperparams(tmpdir): logger = CSVLogger(tmpdir)