tabpfn-time-series 1.0.3__tar.gz → 1.0.5__tar.gz

{tabpfn_time_series-1.0.3 → tabpfn_time_series-1.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tabpfn_time_series
-Version: 1.0.3
+Version: 1.0.5
 Summary: Zero-shot time series forecasting with TabPFNv2
 Project-URL: Homepage, https://github.com/liam-sbhoo/tabpfn-time-series
 Project-URL: Bug Tracker, https://github.com/liam-sbhoo/tabpfn-time-series/issues
@@ -10,6 +10,7 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.10
+Requires-Dist: backoff>=2.2.1
 Requires-Dist: datasets>=4.0
 Requires-Dist: gluonts>=0.16.0
 Requires-Dist: pandas<2.2.0,>=2.1.2
@@ -17,6 +18,7 @@ Requires-Dist: python-dotenv>=1.1.0
 Requires-Dist: pyyaml>=6.0.1
 Requires-Dist: statsmodels>=0.14.5
 Requires-Dist: tabpfn-client>=0.1.7
+Requires-Dist: tabpfn-common-utils[telemetry-interactive]>=0.2.2
 Requires-Dist: tabpfn>=2.0.9
 Requires-Dist: tqdm
 Provides-Extra: dev
@@ -97,3 +99,33 @@ uv pip install -e ".[dev]"
 [![colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/PriorLabs/tabpfn-time-series/blob/main/demo.ipynb)
 
 The demo should explain it all. 😉
+
+## 📊 Anonymous Telemetry
+
+This project collects **anonymous usage telemetry** by default.
+
+The data is used exclusively to help us understand how the library is being used and to guide future improvements.
+
+- **No personal data is collected**
+- **No code, model inputs, or outputs are ever sent**
+- **Data is strictly anonymous and cannot be linked to individuals**
+
+### What we collect
+We only collect high-level, non-identifying information such as:
+- Package version
+- Python version
+- How often fit and inference are called, including simple metadata like the dimensionality of the input and the type of task (e.g., classification vs. regression) (:warning: never the data itself)
+
+See the [Telemetry documentation](https://github.com/priorlabs/tabpfn/blob/main/TELEMETRY.md) for the full details of events and metadata.
+
+This data is processed in compliance with the **General Data Protection Regulation (GDPR)** principles of data minimization and purpose limitation.
+
+For more details, please see our [Privacy Policy](https://priorlabs.ai/privacy_policy/).
+
+### How to opt out
+If you prefer not to send telemetry, you can disable it by setting the following environment variable:
+
+```bash
+export TABPFN_DISABLE_TELEMETRY=1
+```
+---

{tabpfn_time_series-1.0.3 → tabpfn_time_series-1.0.5}/README.md

@@ -62,3 +62,33 @@ uv pip install -e ".[dev]"
 [![colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/PriorLabs/tabpfn-time-series/blob/main/demo.ipynb)
 
 The demo should explain it all. 😉
+
+## 📊 Anonymous Telemetry
+
+This project collects **anonymous usage telemetry** by default.
+
+The data is used exclusively to help us understand how the library is being used and to guide future improvements.
+
+- **No personal data is collected**
+- **No code, model inputs, or outputs are ever sent**
+- **Data is strictly anonymous and cannot be linked to individuals**
+
+### What we collect
+We only collect high-level, non-identifying information such as:
+- Package version
+- Python version
+- How often fit and inference are called, including simple metadata like the dimensionality of the input and the type of task (e.g., classification vs. regression) (:warning: never the data itself)
+
+See the [Telemetry documentation](https://github.com/priorlabs/tabpfn/blob/main/TELEMETRY.md) for the full details of events and metadata.
+
+This data is processed in compliance with the **General Data Protection Regulation (GDPR)** principles of data minimization and purpose limitation.
+
+For more details, please see our [Privacy Policy](https://priorlabs.ai/privacy_policy/).
+
+### How to opt out
+If you prefer not to send telemetry, you can disable it by setting the following environment variable:
+
+```bash
+export TABPFN_DISABLE_TELEMETRY=1
+```
+---

{tabpfn_time_series-1.0.3 → tabpfn_time_series-1.0.5}/demo.ipynb

@@ -35,7 +35,14 @@
    },
    "outputs": [],
    "source": [
-    "# !pip install tabpfn-time-series>=1.0.0"
+    "# !uv pip install tabpfn-time-series>=1.0.3"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "***Note: remember to restart the runtime after the installation.***"
    ]
   },
   {
@@ -1058,18 +1065,6 @@
    "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.12.11"
   }
  },
  "nbformat": 4,

{tabpfn_time_series-1.0.3 → tabpfn_time_series-1.0.5}/pyproject.toml

@@ -24,6 +24,8 @@ dependencies = [
     "datasets>=4.0",
     "python-dotenv>=1.1.0",
     "pyyaml>=6.0.1",
+    "backoff>=2.2.1",
+    "tabpfn-common-utils[telemetry-interactive]>=0.2.2"
 ]
 
 [project.optional-dependencies]

{tabpfn_time_series-1.0.3 → tabpfn_time_series-1.0.5}/tabpfn_time_series/predictor.py

@@ -2,8 +2,13 @@ import logging
 from enum import Enum
 
 from tabpfn_time_series.ts_dataframe import TimeSeriesDataFrame
-from tabpfn_time_series.tabpfn_worker import TabPFNClient, LocalTabPFN, MockTabPFN
+from tabpfn_time_series.tabpfn_worker import (
+    TabPFNClient,
+    LocalTabPFN,
+    MockTabPFN,
+)
 from tabpfn_time_series.defaults import TABPFN_TS_DEFAULT_CONFIG
+from tabpfn_common_utils.telemetry import set_extension
 
 logger = logging.getLogger(__name__)
 
@@ -14,6 +19,7 @@ class TabPFNMode(Enum):
     MOCK = "tabpfn-mock"
 
 
+@set_extension("time-series")
 class TabPFNTimeSeriesPredictor:
     """
     Given a TimeSeriesDataFrame (multiple time series), perform prediction on each time series individually.
@@ -29,6 +35,7 @@ class TabPFNTimeSeriesPredictor:
             TabPFNMode.LOCAL: lambda: LocalTabPFN(config),
             TabPFNMode.MOCK: lambda: MockTabPFN(config),
         }
+        self.tabpfn_mode = tabpfn_mode
         self.tabpfn_worker = worker_mapping[tabpfn_mode]()
 
     def predict(

{tabpfn_time_series-1.0.3 → tabpfn_time_series-1.0.5}/tabpfn_time_series/tabpfn_worker.py

@@ -1,6 +1,8 @@
+import contextvars
 import logging
 from abc import ABC, abstractmethod
-from joblib import Parallel, delayed
+from joblib import Parallel, delayed, parallel_config
+import backoff
 
 from tqdm import tqdm
 import pandas as pd
@@ -14,6 +16,9 @@ from tabpfn_time_series.defaults import TABPFN_TS_DEFAULT_QUANTILE_CONFIG
 
 logger = logging.getLogger(__name__)
 
+# Per-call attempt counter, isolated per thread & task
+_retry_attempts = contextvars.ContextVar("predict_attempts", default=0)
+
 
 class TabPFNWorker(ABC):
     def __init__(
@@ -29,24 +34,7 @@ class TabPFNWorker(ABC):
         train_tsdf: TimeSeriesDataFrame,
         test_tsdf: TimeSeriesDataFrame,
     ):
-        predictions = Parallel(
-            n_jobs=self.num_workers,
-            backend="loky",
-        )(
-            delayed(self._prediction_routine)(
-                item_id,
-                train_tsdf.loc[item_id],
-                test_tsdf.loc[item_id],
-            )
-            for item_id in tqdm(train_tsdf.item_ids, desc="Predicting time series")
-        )
-
-        predictions = pd.concat(predictions)
-
-        # Sort predictions according to original item_ids order (important for MASE and WQL calculation)
-        predictions = predictions.loc[train_tsdf.item_ids]
-
-        return TimeSeriesDataFrame(predictions)
+        raise NotImplementedError("Predict method must be implemented in subclass")
 
     def _prediction_routine(
         self,
@@ -54,8 +42,6 @@ class TabPFNWorker(ABC):
         single_train_tsdf: TimeSeriesDataFrame,
         single_test_tsdf: TimeSeriesDataFrame,
     ) -> pd.DataFrame:
-        # logger.debug(f"Predicting on item_id: {item_id}")
-
         test_index = single_test_tsdf.index
         train_X, train_y = split_time_series_to_X_y(single_train_tsdf.copy())
         test_X, _ = split_time_series_to_X_y(single_test_tsdf.copy())
@@ -116,6 +102,39 @@ class TabPFNWorker(ABC):
         return result
 
 
+def _reset_attempts(_details=None):
+    """Convenience function to reset the attempt counter."""
+    _retry_attempts.set(0)
+
+
+def _predict_giveup_mixed(exc: Exception) -> bool:
+    """Determine whether to give up on a prediction call or not.
+
+    Returns:
+        True if the prediction call should be given up on, False otherwise.
+    """
+    if _is_tabpfn_gcs_429(exc):
+        return False
+
+    # Stop after first retry for non-429
+    return _retry_attempts.get() >= 2
+
+
+def _is_tabpfn_gcs_429(err: Exception) -> bool:
+    """Determine if an error is a 429 error raised from TabPFN API
+    and relates to GCS 429 errors.
+
+    Returns:
+        True if the error is a 429 error raised from TabPFN API.
+    """
+    markers = (
+        "TooManyRequests: 429",
+        "rateLimitExceeded",
+        "cloud.google.com/storage/docs/gcs429",
+    )
+    return any(m in str(err) for m in markers)
+
+
 class TabPFNClient(TabPFNWorker):
     def __init__(
         self,
@@ -135,6 +154,53 @@ class TabPFNClient(TabPFNWorker):
 
         super().__init__(config, num_workers)
 
+    def predict(
+        self,
+        train_tsdf: TimeSeriesDataFrame,
+        test_tsdf: TimeSeriesDataFrame,
+    ):
+        # Run the predictions in parallel
+        with parallel_config(backend="threading"):
+            results = Parallel(
+                n_jobs=self.num_workers,
+            )(
+                delayed(self._prediction_routine)(
+                    item_id,
+                    train_tsdf.loc[item_id],
+                    test_tsdf.loc[item_id],
+                )
+                for item_id in tqdm(train_tsdf.item_ids, desc="Predicting time series")
+            )
+
+        # Convert list to DataFrame
+        predictions = pd.concat(results)
+
+        # Sort predictions according to original item_ids order (important for MASE and WQL calculation)
+        predictions = predictions.loc[train_tsdf.item_ids]
+
+        return TimeSeriesDataFrame(predictions)
+
+    @backoff.on_exception(
+        backoff.expo,
+        Exception,
+        base=1,
+        factor=2,
+        max_tries=5,
+        jitter=backoff.full_jitter,
+        giveup=_predict_giveup_mixed,
+        on_success=_reset_attempts,
+    )
+    def _prediction_routine(
+        self,
+        item_id: str,
+        single_train_tsdf: TimeSeriesDataFrame,
+        single_test_tsdf: TimeSeriesDataFrame,
+    ) -> pd.DataFrame:
+        # Increment attempt count at start of each try
+        _retry_attempts.set(_retry_attempts.get() + 1)
+
+        return super()._prediction_routine(item_id, single_train_tsdf, single_test_tsdf)
+
     def _get_tabpfn_engine(self):
         from tabpfn_client import TabPFNRegressor
 

{tabpfn_time_series-1.0.3 → tabpfn_time_series-1.0.5}/tests/test_predictor.py

@@ -84,7 +84,11 @@ class TestTabPFNTimeSeriesPredictor(unittest.TestCase):
             _ = TabPFNTimeSeriesPredictor(tabpfn_mode=TabPFNMode.LOCAL)
 
     @patch("torch.cuda.is_available", return_value=True)
-    def test_init_with_local_mode_with_gpu(self, mock_is_available):
+    @patch("torch.cuda.device_count", return_value=1)
+    @patch("tabpfn_time_series.tabpfn_worker.LocalTabPFN._download_model")
+    def test_init_with_local_mode_with_gpu(
+        self, mock_download, mock_device_count, mock_is_available
+    ):
         """Test that the predictor initializes with LOCAL mode"""
         predictor = TabPFNTimeSeriesPredictor(tabpfn_mode=TabPFNMode.LOCAL)
         self.assertIsNotNone(predictor.tabpfn_worker)

{tabpfn_time_series-1.0.3 → tabpfn_time_series-1.0.5}/uv.lock

@@ -1,5 +1,5 @@
 version = 1
-revision = 2
+revision = 3
 requires-python = ">=3.10"
 resolution-markers = [
     "python_full_version >= '3.13' and sys_platform == 'linux'",
@@ -256,6 +256,15 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/b7/b8/3fe70c75fe32afc4bb507f75563d39bc5642255d1d94f1f23604725780bf/babel-2.17.0-py3-none-any.whl", hash = "sha256:4d0b53093fdfb4b21c92b5213dba5a1b23885afa8383709427046b21c366e5f2", size = 10182537, upload-time = "2025-02-01T15:17:37.39Z" },
 ]
 
+[[package]]
+name = "backoff"
+version = "2.2.1"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/47/d7/5bbeb12c44d7c4f2fb5b56abce497eb5ed9f34d85701de869acedd602619/backoff-2.2.1.tar.gz", hash = "sha256:03f829f5bb1923180821643f8753b0502c3b682293992485b0eef2807afa5cba", size = 17001, upload-time = "2022-10-05T19:19:32.061Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/df/73/b6e24bd22e6720ca8ee9a85a0c4a2971af8497d8f3193fa05390cbd46e09/backoff-2.2.1-py3-none-any.whl", hash = "sha256:63579f9a0628e06278f7e47b7d7d5b6ce20dc65c5e96a6f3ca99a6adca0396e8", size = 15148, upload-time = "2022-10-05T19:19:30.546Z" },
+]
+
 [[package]]
 name = "backports-tarfile"
 version = "1.2.0"
@@ -612,6 +621,15 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/91/a1/cf2472db20f7ce4a6be1253a81cfdf85ad9c7885ffbed7047fb72c24cf87/distlib-0.3.9-py2.py3-none-any.whl", hash = "sha256:47f8c22fd27c27e25a65601af709b38e4f0a45ea4fc2e710f65755fa8caaaf87", size = 468973, upload-time = "2024-10-09T18:35:44.272Z" },
 ]
 
+[[package]]
+name = "distro"
+version = "1.9.0"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/fc/f8/98eea607f65de6527f8a2e8885fc8015d3e6f5775df186e443e0964a11c3/distro-1.9.0.tar.gz", hash = "sha256:2fa77c6fd8940f116ee1d6b94a2f90b13b5ea8d019b98bc8bafdcabcdd9bdbed", size = 60722, upload-time = "2023-12-24T09:54:32.31Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/12/b3/231ffd4ab1fc9d679809f356cebee130ac7daa00d6d6f3206dd4fd137e9e/distro-1.9.0-py3-none-any.whl", hash = "sha256:7bffd925d65168f85027d8da9af6bddab658135b840670a223589bc0c8ef02b2", size = 20277, upload-time = "2023-12-24T09:54:30.421Z" },
+]
+
 [[package]]
 name = "docutils"
 version = "0.21.2"
@@ -630,6 +648,15 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/87/62/9773de14fe6c45c23649e98b83231fffd7b9892b6cf863251dc2afa73643/einops-0.8.1-py3-none-any.whl", hash = "sha256:919387eb55330f5757c6bea9165c5ff5cfe63a642682ea788a6d472576d81737", size = 64359, upload-time = "2025-02-09T03:17:01.998Z" },
 ]
 
+[[package]]
+name = "eval-type-backport"
+version = "0.2.2"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/30/ea/8b0ac4469d4c347c6a385ff09dc3c048c2d021696664e26c7ee6791631b5/eval_type_backport-0.2.2.tar.gz", hash = "sha256:f0576b4cf01ebb5bd358d02314d31846af5e07678387486e2c798af0e7d849c1", size = 9079, upload-time = "2024-12-21T20:09:46.005Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/ce/31/55cd413eaccd39125368be33c46de24a1f639f2e12349b0361b4678f3915/eval_type_backport-0.2.2-py3-none-any.whl", hash = "sha256:cb6ad7c393517f476f96d456d0412ea80f0a8cf96f6892834cd9340149111b0a", size = 5830, upload-time = "2024-12-21T20:09:44.175Z" },
+]
+
 [[package]]
 name = "exceptiongroup"
 version = "1.3.0"
@@ -662,11 +689,11 @@ wheels = [
 
 [[package]]
 name = "filelock"
-version = "3.18.0"
+version = "3.19.1"
 source = { registry = "https://pypi.org/simple" }
-sdist = { url = "https://files.pythonhosted.org/packages/0a/10/c23352565a6544bdc5353e0b15fc1c563352101f30e24bf500207a54df9a/filelock-3.18.0.tar.gz", hash = "sha256:adbc88eabb99d2fec8c9c1b229b171f18afa655400173ddc653d5d01501fb9f2", size = 18075, upload-time = "2025-03-14T07:11:40.47Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/40/bb/0ab3e58d22305b6f5440629d20683af28959bf793d98d11950e305c1c326/filelock-3.19.1.tar.gz", hash = "sha256:66eda1888b0171c998b35be2bcc0f6d75c388a7ce20c3f3f37aa8e96c2dddf58", size = 17687, upload-time = "2025-08-14T16:56:03.016Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/4d/36/2a115987e2d8c300a974597416d9de88f2444426de9571f4b59b2cca3acc/filelock-3.18.0-py3-none-any.whl", hash = "sha256:c401f4f8377c4464e6db25fff06205fd89bdd83b65eb0488ed1b160f780e21de", size = 16215, upload-time = "2025-03-14T07:11:39.145Z" },
+    { url = "https://files.pythonhosted.org/packages/42/14/42b2651a2f46b022ccd948bca9f2d5af0fd8929c4eec235b8d6d844fbe67/filelock-3.19.1-py3-none-any.whl", hash = "sha256:d38e30481def20772f5baf097c122c3babc4fcdb7e14e57049eb9d88c6dc017d", size = 15988, upload-time = "2025-08-14T16:56:01.633Z" },
 ]
 
 [[package]]
@@ -2173,11 +2200,11 @@ wheels = [
 
 [[package]]
 name = "platformdirs"
-version = "4.3.8"
+version = "4.4.0"
 source = { registry = "https://pypi.org/simple" }
-sdist = { url = "https://files.pythonhosted.org/packages/fe/8b/3c73abc9c759ecd3f1f7ceff6685840859e8070c4d947c93fae71f6a0bf2/platformdirs-4.3.8.tar.gz", hash = "sha256:3d512d96e16bcb959a814c9f348431070822a6496326a4be0911c40b5a74c2bc", size = 21362, upload-time = "2025-05-07T22:47:42.121Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/23/e8/21db9c9987b0e728855bd57bff6984f67952bea55d6f75e055c46b5383e8/platformdirs-4.4.0.tar.gz", hash = "sha256:ca753cf4d81dc309bc67b0ea38fd15dc97bc30ce419a7f58d13eb3bf14c4febf", size = 21634, upload-time = "2025-08-26T14:32:04.268Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/fe/39/979e8e21520d4e47a0bbe349e2713c0aac6f3d853d0e5b34d76206c439aa/platformdirs-4.3.8-py3-none-any.whl", hash = "sha256:ff7059bb7eb1179e2685604f4aaf157cfd9535242bd23742eadc3c13542139b4", size = 18567, upload-time = "2025-05-07T22:47:40.376Z" },
+    { url = "https://files.pythonhosted.org/packages/40/4b/2028861e724d3bd36227adfa20d3fd24c3fc6d52032f4a93c133be5d17ce/platformdirs-4.4.0-py3-none-any.whl", hash = "sha256:abd01743f24e5287cd7a5db3752faf1a2d65353f38ec26d98e25a6db65958c85", size = 18654, upload-time = "2025-08-26T14:32:02.735Z" },
 ]
 
 [[package]]
@@ -2189,6 +2216,23 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/54/20/4d324d65cc6d9205fabedc306948156824eb9f0ee1633355a8f7ec5c66bf/pluggy-1.6.0-py3-none-any.whl", hash = "sha256:e920276dd6813095e9377c0bc5566d94c932c33b27a3e3945d8389c374dd4746", size = 20538, upload-time = "2025-05-15T12:30:06.134Z" },
 ]
 
+[[package]]
+name = "posthog"
+version = "6.7.5"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "backoff" },
+    { name = "distro" },
+    { name = "python-dateutil" },
+    { name = "requests" },
+    { name = "six" },
+    { name = "typing-extensions" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/1f/f3/d9055bcd190980730bdc600318b34290e85fcb0afb1e31f83cdc33f92615/posthog-6.7.5.tar.gz", hash = "sha256:f4f32b4a4b0df531ae8f80f255a33a49e8880c8c1b62712e6b640535e33a905f", size = 118558, upload-time = "2025-09-16T12:40:34.431Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/42/b4/b40f8467252b4ff481e54a9767b211b4ff83114e6d0b6f481852d0ef3e46/posthog-6.7.5-py3-none-any.whl", hash = "sha256:95b00f915365939e63fa183635bad1caaf89cf4a24b63c8bb6983f2a22a56cb3", size = 136766, upload-time = "2025-09-16T12:40:32.741Z" },
+]
+
 [[package]]
 name = "pre-commit"
 version = "4.2.0"
@@ -2556,6 +2600,20 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/32/56/8a7ca5d2cd2cda1d245d34b1c9a942920a718082ae8e54e5f3e5a58b7add/pydantic_core-2.33.2-pp311-pypy311_pp73-win_amd64.whl", hash = "sha256:329467cecfb529c925cf2bbd4d60d2c509bc2fb52a20c1045bf09bb70971a9c1", size = 2066757, upload-time = "2025-04-23T18:33:30.645Z" },
 ]
 
+[[package]]
+name = "pydantic-settings"
+version = "2.10.1"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "pydantic" },
+    { name = "python-dotenv" },
+    { name = "typing-inspection" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/68/85/1ea668bbab3c50071ca613c6ab30047fb36ab0da1b92fa8f17bbc38fd36c/pydantic_settings-2.10.1.tar.gz", hash = "sha256:06f0062169818d0f5524420a360d632d5857b83cffd4d42fe29597807a1614ee", size = 172583, upload-time = "2025-06-24T13:26:46.841Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/58/f0/427018098906416f580e3cf1366d3b1abfb408a0652e9f31600c24a1903c/pydantic_settings-2.10.1-py3-none-any.whl", hash = "sha256:a60952460b99cf661dc25c29c0ef171721f98bfcb52ef8d9ea4c943d7c8cc796", size = 45235, upload-time = "2025-06-24T13:26:45.485Z" },
+]
+
 [[package]]
 name = "pygments"
 version = "2.19.2"
@@ -2806,7 +2864,7 @@ wheels = [
 
 [[package]]
 name = "requests"
-version = "2.32.4"
+version = "2.32.5"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "certifi" },
@@ -2814,9 +2872,9 @@ dependencies = [
     { name = "idna" },
     { name = "urllib3" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/e1/0a/929373653770d8a0d7ea76c37de6e41f11eb07559b103b1c02cafb3f7cf8/requests-2.32.4.tar.gz", hash = "sha256:27d0316682c8a29834d3264820024b62a36942083d52caf2f14c0591336d3422", size = 135258, upload-time = "2025-06-09T16:43:07.34Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/c9/74/b3ff8e6c8446842c3f5c837e9c3dfcfe2018ea6ecef224c710c85ef728f4/requests-2.32.5.tar.gz", hash = "sha256:dbba0bac56e100853db0ea71b82b4dfd5fe2bf6d3754a8893c3af500cec7d7cf", size = 134517, upload-time = "2025-08-18T20:46:02.573Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/7c/e4/56027c4a6b4ae70ca9de302488c5ca95ad4a39e190093d6c1a8ace08341b/requests-2.32.4-py3-none-any.whl", hash = "sha256:27babd3cda2a6d50b30443204ee89830707d396671944c998b5975b031ac2b2c", size = 64847, upload-time = "2025-06-09T16:43:05.728Z" },
+    { url = "https://files.pythonhosted.org/packages/1e/db/4254e3eabe8020b458f1a747140d32277ec7a271daf1d235b70dc0b4e6e3/requests-2.32.5-py3-none-any.whl", hash = "sha256:2462f94637a34fd532264295e186976db0f5d453d1cdd31473c85a6a161affb6", size = 64738, upload-time = "2025-08-18T20:46:00.542Z" },
 ]
 
 [[package]]
@@ -3357,25 +3415,30 @@ wheels = [
 
 [[package]]
 name = "tabpfn"
-version = "2.0.9"
+version = "2.2.1"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "einops" },
+    { name = "eval-type-backport" },
     { name = "huggingface-hub" },
+    { name = "joblib" },
     { name = "pandas" },
+    { name = "pydantic" },
+    { name = "pydantic-settings" },
     { name = "scikit-learn" },
     { name = "scipy" },
+    { name = "tabpfn-common-utils" },
     { name = "torch" },
     { name = "typing-extensions" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/cb/32/a705d6b426800e79338ed641cb7d96877e069289309455cb215df14126b6/tabpfn-2.0.9.tar.gz", hash = "sha256:7e6e006a61e8b08011c1a6761eef3c514c4ff16f5698ed655443c0704a3f93a6", size = 139991, upload-time = "2025-05-07T13:21:36.067Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/e5/8b/ab7e3c71fc5fb6be61295ca1fb880b09df567257e9e4228d77cd79a4e1ec/tabpfn-2.2.1.tar.gz", hash = "sha256:706c03c3d3dc478118561761c29fa95727b6e239671922c4c40a8e4b3cc28fbf", size = 200487, upload-time = "2025-09-17T10:42:15.649Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/35/f8/b0e7ff83484acfbe670b2f56c28c6bcf09ae84284e624cd45755a8b7073c/tabpfn-2.0.9-py3-none-any.whl", hash = "sha256:04e3bb989e9328d510ea4fccb6c6a36c8269630685d460af4df5eb268206bf21", size = 128881, upload-time = "2025-05-07T13:21:34.227Z" },
+    { url = "https://files.pythonhosted.org/packages/91/e9/085c60d5e447083ac6705b7b8bfac90773eb4191c00ddc70607f4d8b339b/tabpfn-2.2.1-py3-none-any.whl", hash = "sha256:9aa3a0df8ecebcae5f426b1fec9bb2e2a690880bc8ad079191e8d959bf0f157b", size = 173723, upload-time = "2025-09-17T10:42:14.435Z" },
 ]
 
 [[package]]
 name = "tabpfn-client"
-version = "0.1.7"
+version = "0.1.9"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "httpx" },
@@ -3385,17 +3448,38 @@ dependencies = [
     { name = "scikit-learn" },
     { name = "sseclient-py" },
     { name = "tqdm" },
+    { name = "typing-extensions" },
     { name = "xxhash" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/e2/cc/97be739da8d4113c13b8c9fa031908d59210005d38f944e0d678717cf91c/tabpfn_client-0.1.7.tar.gz", hash = "sha256:f8b82318ce22240b7622a05bea2422204ddd5374df335a19284eea0e2252924e", size = 17066065, upload-time = "2025-03-18T12:58:03.737Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/50/8f/1bbca5db6561bc54f91ce41de99835a962a7a16c53bcfae0a7c5593f16a2/tabpfn_client-0.1.9.tar.gz", hash = "sha256:b68b27154686c37119a53b5063de79b45f4c0074fddf47b8813c5cd738b723cf", size = 1877401, upload-time = "2025-07-28T15:09:43.134Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/67/a5/82be418a16d7330eac880659b1ce7fbcff5c976ff79240b1446235e4ed40/tabpfn_client-0.1.9-py3-none-any.whl", hash = "sha256:51826f3a464b2d52d9c2772a6171c55a4c0ab22d9693f5bd2eb9fd5b0fd24640", size = 1895907, upload-time = "2025-07-28T15:09:41.605Z" },
+]
+
+[[package]]
+name = "tabpfn-common-utils"
+version = "0.2.2"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "filelock" },
+    { name = "numpy" },
+    { name = "pandas" },
+    { name = "platformdirs" },
+    { name = "posthog" },
+    { name = "requests" },
+    { name = "scikit-learn" },
+    { name = "typing-extensions" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/cf/0a/dae8031fd9a478c0b6b682f1585d2e4e288727f569eb3316af3b981d1ec0/tabpfn_common_utils-0.2.2.tar.gz", hash = "sha256:d4a3b9acd7d36502ac75bfce549acda26aafbe9c1e7617052fbe1932c211d174", size = 1927582, upload-time = "2025-10-02T11:54:59.704Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/5b/55/ba50a32b222616eb5653958fc94d91f54ab74109417ada52bb2aecb89a15/tabpfn_client-0.1.7-py3-none-any.whl", hash = "sha256:84245c893b06aa084795c31c04ec3f58186499dfcd7f6f2b8821f9a0a6028136", size = 1902270, upload-time = "2025-03-18T12:57:52.842Z" },
+    { url = "https://files.pythonhosted.org/packages/7a/be/6307e7f8c55963aa74c72dbbe0e360b62a688a484f73227adf7cc1ddde02/tabpfn_common_utils-0.2.2-py3-none-any.whl", hash = "sha256:9ac650c252be49c34d12d210eceeeb3d9cdc7ac5396df7426e5b86b93ae5429e", size = 32104, upload-time = "2025-10-02T11:54:57.702Z" },
 ]
 
 [[package]]
 name = "tabpfn-time-series"
 source = { editable = "." }
 dependencies = [
+    { name = "backoff" },
     { name = "datasets" },
     { name = "gluonts" },
     { name = "pandas" },
@@ -3404,6 +3488,7 @@ dependencies = [
     { name = "statsmodels" },
     { name = "tabpfn" },
     { name = "tabpfn-client" },
+    { name = "tabpfn-common-utils" },
     { name = "tqdm" },
 ]
 
@@ -3424,6 +3509,7 @@ dev = [
 
 [package.metadata]
 requires-dist = [
+    { name = "backoff", specifier = ">=2.2.1" },
     { name = "build", marker = "extra == 'dev'" },
     { name = "datasets", specifier = ">=4.0" },
     { name = "gluonts", specifier = ">=0.16.0" },
@@ -3441,6 +3527,7 @@ requires-dist = [
     { name = "submitit", marker = "extra == 'dev'", specifier = ">=1.5.2" },
     { name = "tabpfn", specifier = ">=2.0.9" },
     { name = "tabpfn-client", specifier = ">=0.1.7" },
+    { name = "tabpfn-common-utils", extras = ["telemetry-interactive"], specifier = ">=0.2.2" },
     { name = "tqdm" },
     { name = "twine", marker = "extra == 'dev'" },
     { name = "wandb", marker = "extra == 'dev'", specifier = ">=0.19.8" },