Merge pull request #71 from PriorLabs/disable-pseudo-tracking

Drop install ID tracking

Author: safaricd

src/tabpfn_common_utils/telemetry/core/events.py

@@ -16,7 +16,6 @@
     nvmlShutdown,
 )
 from .runtime import get_execution_context
-from .state import get_property, set_property
 
 
 def _uuid4() -> str:
@@ -256,50 +255,6 @@ def _get_runtime_environment() -> Optional[str]:
     return exec_context.environment
 
 
-@lru_cache(maxsize=1)
-def _get_install_id() -> str:
-    """Get or create the install ID. If not set in disk-cached
-    state, generate a new one and store it in the state.
-
-    Returns:
-        str: The install ID.
-    """
-    install_id = get_property("install_id")
-
-    # Fallback to new install ID
-    if install_id is None:
-        install_id = _uuid4()
-        set_property("install_id", install_id)
-
-    return install_id
-
-
-@lru_cache(maxsize=1)
-def _get_install_date() -> str:
-    """Return the install date as YYYY-MM-DD.
-
-    In case the user was using TabPFN before this change was made,
-    the install date will not correspond to the actual install date,
-    but rather when the user upgraded the page and made a first call.
-
-    Returns:
-        str: The install date as YYYY-MM-DD.
-    """
-    value = get_property("install_date")
-
-    if isinstance(value, str):
-        try:
-            dt = datetime.fromisoformat(value.replace("Z", "+00:00"))
-            return dt.date().isoformat()
-        except ValueError:
-            pass
-
-    # Fallback to current date
-    value = _utc_now_date()
-    set_property("install_date", value)
-    return value
-
-
 @dataclass
 class BaseTelemetryEvent:
     """
@@ -355,12 +310,6 @@ class SessionEvent(BaseTelemetryEvent):
     is a single init call to TabPFNClassifier or TabPFNRegressor.
     """
 
-    # Install ID of the user
-    install_id: str = field(default_factory=_get_install_id, init=False)
-
-    # Install date of the user
-    install_date: str = field(default_factory=_get_install_date, init=False)
-
     @property
     def name(self) -> str:
         return "session"
@@ -375,9 +324,6 @@ class ModelLoadEvent(BaseTelemetryEvent):
     # Status of the model download attempt
     status: Literal["success", "failed"]
 
-    # Install ID of the user
-    install_id: str = field(default_factory=_get_install_id, init=False)
-
     # Name of the model, may be a HuggingFace repo ID
     model_name: Optional[str] = field(default=None)
 
@@ -439,9 +385,6 @@ class ModelCallEvent(BaseTelemetryEvent):
     # Task associated with the model call
     task: Literal["classification", "regression"]
 
-    # Install ID of the user
-    install_id: str = field(default_factory=_get_install_id, init=False)
-
     # Version of the PyTorch
     torch_version: str = field(default_factory=_get_torch_version, init=False)
 

src/tabpfn_common_utils/telemetry/core/state.py

@@ -31,10 +31,6 @@
 _DEFAULT_STATE: dict[str, Any] = {
     # Date and time when state file was created
     "created_at": None,
-    # Anonymous user ID, only with consent
-    "user_id": None,
-    # Email address, only with consent
-    "email": None,
     # Number of times prompts were shown
     "nr_prompts": 0,
     # Date and time when last prompt was shown

src/tabpfn_common_utils/telemetry/interactive/flows.py

@@ -5,7 +5,6 @@
 from datetime import datetime, timedelta, timezone
 
 from typing import List
-from ..core.config import download_config
 from ..core.state import get_property, set_property
 from ..core import PingEvent, SessionEvent, capture_event
 from .prompts.base import PromptSpec
@@ -76,52 +75,6 @@ def _trigger_prompts(delta_days: int, max_prompts: int) -> bool:
     Returns:
         True if a prompt should be shown, False otherwise.
     """
-    utc_now = datetime.now(timezone.utc)
-
-    # Download prompt configuration
-    config = download_config()
-
-    # By default, don't prompt
-    if not config.get("prompt_user", False):
-        return False
-
-    # If new installation, don't prompt
-    install_date = get_property("install_date", data_type=datetime)
-    if not install_date:
-        set_property("install_date", utc_now)
-        return False
-
-    # Avoid prompt in first 24 hours
-    delta_hours = config.get("prompt_delta_hours", 24)
-
-    # Ensure timezone-aware datetime
-    if install_date.tzinfo is None:
-        install_date = install_date.replace(tzinfo=timezone.utc)
-
-    if utc_now - install_date <= timedelta(hours=delta_hours):
-        return False
-
-    # If used <= 5 times, don't prompt
-    nr_usages = get_property("nr_usages", 0, data_type=int)
-    set_property("nr_usages", nr_usages + 1)
-
-    if nr_usages < config.get("prompt_nr_usages", 5):
-        return False
-
-    # If last prompted > 30 days, prompt
-    last_prompted_at = get_property("last_prompted_at", data_type=datetime)
-    if not last_prompted_at:
-        return True
-
-    # Check if the maximum number of prompts has been reached
-    nr_prompts = get_property("nr_prompts", 0, data_type=int)
-    if nr_prompts >= max_prompts:
-        return False
-
-    # Check if the time since the last prompt is greater than the delta days
-    if utc_now - last_prompted_at >= timedelta(days=delta_days):
-        return True
-
     return False
 
 

src/tabpfn_common_utils/telemetry/interactive/prompts/identity.py

@@ -1,10 +1,8 @@
 import logging
-import uuid
 import textwrap
 
 from typing import Any, Dict, Optional
 from .base import render_html, PromptResult, Outcome, parse_input, PromptSpec
-from ...core.state import get_property, set_property
 
 
 # Logger
@@ -69,8 +67,7 @@ def _should_prompt() -> bool:
     """
     Check if the user should be prompted to subscribe to the newsletter.
     """
-    user_id = get_property("user_id")
-    return user_id is None
+    return False
 
 
 def _on_done(res: PromptResult) -> None:
@@ -79,11 +76,7 @@ def _on_done(res: PromptResult) -> None:
     Args:
         res: The prompt result.
     """
-    if res.outcome != "accepted":
-        return
-
-    user_id = str(uuid.uuid4())
-    set_property("user_id", user_id)
+    return None
 
 
 class IdentityPrompt:

src/tabpfn_common_utils/telemetry/interactive/prompts/newsletter.py

@@ -2,11 +2,9 @@
 import os
 import re
 
-import requests
 import textwrap
 from typing import Any, Dict, Optional
 from .base import render_html, PromptResult, Outcome, parse_input, PromptSpec
-from ...core.state import get_property, set_property
 
 
 # Logger
@@ -90,19 +88,14 @@ def _subscribe_user(email: str) -> None:
     Args:
         email: The email of the user to opt_in.
     """
-    endpoint = _API_URL + "/newsletter/subscribe"
-    r = requests.post(endpoint, json={"email": email.lower()}, timeout=5)
-    if r.status_code != 200:
-        _logger.debug(f"Failed to subscribe user {email}: {r.text}")
-        return
+    return None
 
 
 def _should_prompt() -> bool:
     """
     Check if the user should be prompted to subscribe to the newsletter.
     """
-    email = get_property("email")
-    return email is None
+    return False
 
 
 def _on_done(res: PromptResult) -> None:
@@ -111,22 +104,7 @@ def _on_done(res: PromptResult) -> None:
     Args:
         res: The prompt result.
     """
-    if res.outcome != "accepted":
-        return
-
-    payload = res.data or {}
-    email = payload.get("email")
-    if not email:
-        return
-
-    try:
-        # Subscribe the user to the newsletter using the API
-        _subscribe_user(email)
-    except Exception:
-        # best-effort; don't disrupt host process
-        pass
-
-    set_property("email", email)
+    return None
 
 
 class NewsletterPrompt:

tests/telemetry/core/test_events.py

@@ -402,7 +402,6 @@ def test_model_load_event_inherits_base_properties(self):
         assert isinstance(event.tabpfn_version, str)
         assert isinstance(event.timestamp, datetime)
         assert event.source == "sdk"
-        assert isinstance(event.install_id, str)
 
     def test_model_load_event_properties_method(self):
         """Test ModelLoadEvent properties method"""
@@ -420,7 +419,6 @@ def test_model_load_event_properties_method(self):
         assert props["failure_reason"] == "Download timeout"
         assert "python_version" in props
         assert "tabpfn_version" in props
-        assert "install_id" in props
 
 
 class TestExtensionEntryEvent:

tests/telemetry/core/test_state.py

@@ -41,8 +41,6 @@ def test_filelock_availability(self):
     def test_default_state_schema(self):
         assert isinstance(_DEFAULT_STATE, dict)
         assert "created_at" in _DEFAULT_STATE
-        assert "user_id" in _DEFAULT_STATE
-        assert "email" in _DEFAULT_STATE
         assert "nr_prompts" in _DEFAULT_STATE
         assert "last_prompted_at" in _DEFAULT_STATE
 
@@ -311,7 +309,6 @@ def test_load_state_non_existing_file(self):
             with patch(name, return_value=non_existent_path):
                 result = load_state()
                 assert "created_at" in result
-                assert result["user_id"] is None
 
 
 class TestSaveState: