diff --git a/src/axolotl/telemetry/manager.py b/src/axolotl/telemetry/manager.py index 82d310cdc..46ef389aa 100644 --- a/src/axolotl/telemetry/manager.py +++ b/src/axolotl/telemetry/manager.py @@ -5,7 +5,6 @@ import importlib import logging import os import platform -import time import uuid from pathlib import Path from typing import Any @@ -20,21 +19,6 @@ LOG = logging.getLogger(__name__) POSTHOG_HOST = "https://app.posthog.com" POSTHOG_WRITE_KEY = "phc_1kUR0o04oJKKTTeSsIz2Mfm5mpiVsQEf2WOlzljMD7y" -OPT_OUT_WARNING_SLEEP_SECONDS = 10 -OPT_OUT_WARNING = ( - "\nTelemetry is now enabled by default to help improve Axolotl. " - "If you'd like to disable it, set AXOLOTL_DO_NOT_TRACK=1 in your environment.\n\n" - "Telemetry data helps us understand:\n" - "- Which features are most used\n" - "- What hardware configurations to prioritize\n" - "- Where users encounter errors\n\n" - "Personally identifiable information (PII) is not collected.\n\n" - "To remove this warning, explicitly set AXOLOTL_DO_NOT_TRACK=0 (enable telemetry) " - "or AXOLOTL_DO_NOT_TRACK=1 (disable telemetry).\n\n" - "For details, see: https://docs.axolotl.ai/docs/telemetry.html\n\n" - f"Sleeping for {OPT_OUT_WARNING_SLEEP_SECONDS}s..." -) - WHITELIST_PATH = str(Path(__file__).parent / "whitelist.yaml") # NOTE: Need to keep these up to date with any config schema changes @@ -183,11 +167,6 @@ class TelemetryManager: "false", "true", ): - # Print opt-out info message for main process only - if is_main_process(): - LOG.warning(OPT_OUT_WARNING) - time.sleep(OPT_OUT_WARNING_SLEEP_SECONDS) - return True # Only rank 0 will send telemetry diff --git a/tests/telemetry/test_manager.py b/tests/telemetry/test_manager.py index 2eeae2f11..36ca44c35 100644 --- a/tests/telemetry/test_manager.py +++ b/tests/telemetry/test_manager.py @@ -118,20 +118,6 @@ def test_telemetry_disabled_for_non_main_process(telemetry_manager_class): assert not manager.enabled -def test_opt_in_info_displayed(telemetry_manager_class): - """Test that opt-in info is displayed when telemetry is not configured""" - with ( - patch.dict(os.environ, {"RANK": "0"}, clear=True), - patch("logging.Logger.warning") as mock_warning, - patch("time.sleep"), - ): - telemetry_manager_class() - assert any( - "Telemetry is now enabled by default" in str(call) - for call in mock_warning.call_args_list - ) - - def test_is_whitelisted(telemetry_manager_class, mock_whitelist): """Test org whitelist functionality""" with (