sleep on all ranks in distributed setting

2025-02-24 00:53:58 +00:00
parent ef4990f304
commit 8b75205d3b
2 changed files with 13 additions and 10 deletions
--- a/tests/telemetry/test_manager.py
+++ b/tests/telemetry/test_manager.py
@@ -76,7 +76,9 @@ def test_telemetry_disabled_with_do_not_track(telemetry_manager_class):

 def test_telemetry_disabled_for_non_main_process(telemetry_manager_class):
    """Test that telemetry is disabled for non-main processes"""
-    with patch("axolotl.telemetry.manager.is_main_process", return_value=False):
+    with patch.dict(os.environ, {"AXOLOTL_DO_NOT_TRACK": "0"}), patch(
+        "axolotl.telemetry.manager.is_main_process", return_value=False
+    ):
        manager = telemetry_manager_class()
        assert not manager.enabled