diff --git a/tests/train/test_tracking.py b/tests/train/test_tracking.py index 559c05abea..7018ea920e 100644 --- a/tests/train/test_tracking.py +++ b/tests/train/test_tracking.py @@ -14,7 +14,7 @@ def test_wandb_init_receives_tags(): Tracking( project_name="proj", experiment_name="exp", - backends=["wandb"], + backend="wandb", config={}, tags=["foo", "bar"], ) @@ -33,7 +33,7 @@ def test_wandb_init_tags_default_none(): Tracking( project_name="proj", experiment_name="exp", - backends=["wandb"], + backend="wandb", config={}, )