diff --git a/src/tabpfn_common_utils/telemetry/core/events.py b/src/tabpfn_common_utils/telemetry/core/events.py index c01750b..b2c83d4 100644 --- a/src/tabpfn_common_utils/telemetry/core/events.py +++ b/src/tabpfn_common_utils/telemetry/core/events.py @@ -439,6 +439,9 @@ 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) diff --git a/src/tabpfn_common_utils/telemetry/core/runtime.py b/src/tabpfn_common_utils/telemetry/core/runtime.py index 5f720ae..663c673 100644 --- a/src/tabpfn_common_utils/telemetry/core/runtime.py +++ b/src/tabpfn_common_utils/telemetry/core/runtime.py @@ -19,6 +19,9 @@ "aws", "azure", "databricks", + "modal", + "mlflow", + "kubernetes", ] # Static list of environment hints, purely heuristic based on env variables. @@ -50,6 +53,15 @@ "DATABRICKS_WORKSPACE_URL", "DB_IS_DRIVER", ], + "modal": ["MODAL_CLOUD_PROVIDER", "MODAL_IMAGE_ID", "MODAL_REGION"], + "mlflow": [ + "MLFLOW_RUN_ID", + "MLFLOW_RUN_NAME", + ], + "kubernetes": [ + "KUBERNETES_SERVICE_HOST", + "KUBERNETES_POD_NAME", + ], # Cloud providers "aws": [ # Generic AWS environment hints