Skip to content

Commit 834bca5

Browse files
committed
change test
1 parent ff869e1 commit 834bca5

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tests/loggers/test_wandb.py

+6-3
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,12 @@ def test_wandb_logger(wandb):
1919
logger.log_metrics({'acc': 1.0}, step=3)
2020
wandb.init().log.assert_called_once_with({'global_step': 3, 'acc': 1.0})
2121

22-
logger.log_hyperparams({'test': None})
23-
wandb.init().config.update.assert_called_once_with({'test': None}, allow_val_change=True)
24-
22+
logger.log_hyperparams({'test': None, 'nested': {'a': 1}, 'b': [2, 3, 4]})
23+
wandb.init().config.update.assert_called_once_with(
24+
{'test': 'None', 'nested/a': 1, 'b': [2, 3, 4]},
25+
allow_val_change=True,
26+
)
27+
2528
logger.watch('model', 'log', 10)
2629
wandb.init().watch.assert_called_once_with('model', log='log', log_freq=10)
2730

0 commit comments

Comments
 (0)