lint

intelligent-machine-learning · Jan 2, 2025 · 6014f5a · 6014f5a
1 parent 2cb2c45
commit 6014f5a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dlrover/python/elastic_agent/diagnosis/diagnosis_agent.py b/dlrover/python/elastic_agent/diagnosis/diagnosis_agent.py
@@ -246,7 +246,10 @@ def send_heartbeat(self):
             logger.warning(f"fail to report a heartbeat: {e}")
 
     def _periodically_report(self):
-        logger.info("Start diagnosis agent reporter.")
+        logger.info("Start diagnosis agent periodically reporter.")
         while True:
+            if self._stopped:
+                logger.info("Stop periodically reporter.")
+                break
             self.send_heartbeat()
             time.sleep(15)