From 834bca59c9292e5da6c270808c05d812362ba836 Mon Sep 17 00:00:00 2001 From: Anthony Bisulco Date: Tue, 7 Jul 2020 20:41:59 -0400 Subject: [PATCH] change test --- tests/loggers/test_wandb.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tests/loggers/test_wandb.py b/tests/loggers/test_wandb.py index aa8b616bcf475..182b518d1e52d 100644 --- a/tests/loggers/test_wandb.py +++ b/tests/loggers/test_wandb.py @@ -19,9 +19,12 @@ def test_wandb_logger(wandb): logger.log_metrics({'acc': 1.0}, step=3) wandb.init().log.assert_called_once_with({'global_step': 3, 'acc': 1.0}) - logger.log_hyperparams({'test': None}) - wandb.init().config.update.assert_called_once_with({'test': None}, allow_val_change=True) - + logger.log_hyperparams({'test': None, 'nested': {'a': 1}, 'b': [2, 3, 4]}) + wandb.init().config.update.assert_called_once_with( + {'test': 'None', 'nested/a': 1, 'b': [2, 3, 4]}, + allow_val_change=True, + ) + logger.watch('model', 'log', 10) wandb.init().watch.assert_called_once_with('model', log='log', log_freq=10)