temporarily removing flag check in unit test for system_metrics_monit…

…or build
mosaicml · Jun 7, 2024 · b9df1ed · b9df1ed
1 parent 31c904b
commit b9df1ed
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/callbacks/test_system_metrics_monitor.py b/tests/callbacks/test_system_metrics_monitor.py
@@ -6,11 +6,9 @@
 
 
 def test_system_metrics_monitor_callback_builds():
-    kwargs = {'log_all_data': True}
     callback = build_callback(
         'system_metrics_monitor',
-        kwargs=kwargs,
+        kwargs={},
         train_config={'train_loader': {}},
     )
     assert isinstance(callback, SystemMetricsMonitor)
-    assert getattr(callback, 'log_all_data', None) is True