tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: do not push telemetry on non-master rank (#3438)

Browse Source
This commit is contained in:
NanoCode012
2026-03-02 15:31:20 +07:00
committed by GitHub
parent 7f23b302d1
commit f447bce1db
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/axolotl/telemetry/manager.py
View File

@@ -156,6 +156,10 @@ class TelemetryManager:
Returns: Returns:
Boolean denoting whether telemetry is enabled or not. Boolean denoting whether telemetry is enabled or not.
""" """
# Only rank 0 will send telemetry
if not is_main_process():
return False
# Parse relevant env vars # Parse relevant env vars
axolotl_do_not_track = os.getenv("AXOLOTL_DO_NOT_TRACK") axolotl_do_not_track = os.getenv("AXOLOTL_DO_NOT_TRACK")
do_not_track = os.getenv("DO_NOT_TRACK") do_not_track = os.getenv("DO_NOT_TRACK")
@@ -169,10 +173,6 @@ class TelemetryManager:
): ):
return True return True
# Only rank 0 will send telemetry
if not is_main_process():
return False
if do_not_track is None: if do_not_track is None:
do_not_track = "0" do_not_track = "0"