From b9df1ed47fc58045b9d1396652f06b9aa220761e Mon Sep 17 00:00:00 2001
From: JackZ-db <jack.zhang@databricks.com>
Date: Fri, 7 Jun 2024 14:03:43 -0700
Subject: [PATCH] temporarily removing flag check in unit test for
 system_metrics_monitor build

---
 tests/callbacks/test_system_metrics_monitor.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/tests/callbacks/test_system_metrics_monitor.py b/tests/callbacks/test_system_metrics_monitor.py
index 4a7459b0d5..5d2ccbb484 100644
--- 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