fix refactor mistakes

CrayLabs · Feb 22, 2024 · d8aa972 · d8aa972
1 parent cd12d30
commit d8aa972
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/smartsim/_core/entrypoints/telemetrymonitor.py b/smartsim/_core/entrypoints/telemetrymonitor.py
@@ -857,7 +857,6 @@ def get_parser() -> argparse.ArgumentParser:
     parser = get_parser()
     args = parser.parse_args()
 
-    # log = logging.getLogger(f"{__name__}.TelemetryMonitor")
     logger.setLevel(logging.DEBUG)
     logger.propagate = False
 

diff --git a/tests/test_collectors.py b/tests/test_collectors.py
@@ -209,7 +209,7 @@ async def test_dbconncollector_integration(mock_entity, local_db):
     output data matches expectations and proper db client API uage"""
     entity = mock_entity(port=local_db.ports[0])
 
-    collector = DbConnectionCollector(entity)
+    collector = DbConnectionCollector(entity, LogSink())
 
     await collector.prepare()
     await collector.collect()

diff --git a/tests/test_telemetry_monitor.py b/tests/test_telemetry_monitor.py
@@ -411,7 +411,7 @@ def is_alive(self) -> bool:
     duration = 2
 
     ts0 = get_ts()
-    event_loop(observer, mani_handler, frequency, logger, cooldown_duration=duration)
+    event_loop(observer, mani_handler, frequency, duration)
     ts1 = get_ts()
 
     assert ts1 - ts0 >= duration
@@ -423,7 +423,7 @@ def is_alive(self) -> bool:
     duration = 5
 
     ts0 = get_ts()
-    event_loop(observer, mani_handler, frequency, logger, cooldown_duration=duration)
+    event_loop(observer, mani_handler, frequency,  duration)
     ts1 = get_ts()
 
     assert ts1 - ts0 >= duration