diff --git a/tests/callbacks/callback_settings.py b/tests/callbacks/callback_settings.py index 279f47a695..4b3ef478a2 100644 --- a/tests/callbacks/callback_settings.py +++ b/tests/callbacks/callback_settings.py @@ -146,6 +146,11 @@ NeptuneLogger: { 'mode': 'debug', }, + WandBLogger: { + 'init_kwargs': { + 'mode': 'offline', + }, + }, composer.profiler.Profiler: { 'trace_handlers': [MagicMock()], 'schedule': composer.profiler.cyclic_schedule(),