tabpfn-time-series 1.0.1__py3-none-any.whl → 1.0.2__py3-none-any.whl

tabpfn_time_series/__init__.py

@@ -1,21 +1,20 @@
-from pathlib import Path
+from importlib import metadata
 
 try:
-    import tomllib  # Python 3.11+
-except ImportError:
-    import tomli as tomllib  # Python <3.11, requires 'tomli' package
-
-with (Path(__file__).parent.parent / "pyproject.toml").open("rb") as f:
-    __version__ = tomllib.load(f)["project"]["version"]
-
+    __version__ = metadata.version("tabpfn-time-series")
+except metadata.PackageNotFoundError:
+    # package is not installed from PyPI (e.g. from source)
+    __version__ = "0.0.0"
 
 from .features import FeatureTransformer
 from .predictor import TabPFNTimeSeriesPredictor, TabPFNMode
 from .defaults import TABPFN_TS_DEFAULT_QUANTILE_CONFIG
+from .ts_dataframe import TimeSeriesDataFrame
 
 __all__ = [
     "FeatureTransformer",
     "TabPFNTimeSeriesPredictor",
     "TabPFNMode",
     "TABPFN_TS_DEFAULT_QUANTILE_CONFIG",
+    "TimeSeriesDataFrame",
 ]

tabpfn_time_series/data_preparation.py

@@ -2,7 +2,7 @@ import pandas as pd
 import numpy as np
 
 import datasets
-from autogluon.timeseries import TimeSeriesDataFrame
+from tabpfn_time_series.ts_dataframe import TimeSeriesDataFrame
 
 
 def generate_test_X(

tabpfn_time_series/features/feature_transformer.py

@@ -2,7 +2,7 @@ from typing import List, Tuple
 
 import pandas as pd
 
-from autogluon.timeseries import TimeSeriesDataFrame
+from tabpfn_time_series.ts_dataframe import TimeSeriesDataFrame
 from tabpfn_time_series.features.feature_generator_base import (
     FeatureGenerator,
 )

tabpfn_time_series/plot.py

@@ -2,7 +2,7 @@ import numpy as np
 import pandas as pd
 import matplotlib.pyplot as plt
 
-from autogluon.timeseries import TimeSeriesDataFrame
+from tabpfn_time_series.ts_dataframe import TimeSeriesDataFrame
 
 
 def is_subset(tsdf_A: TimeSeriesDataFrame, tsdf_B: TimeSeriesDataFrame) -> bool:

tabpfn_time_series/predictor.py

@@ -1,8 +1,7 @@
 import logging
 from enum import Enum
 
-from autogluon.timeseries import TimeSeriesDataFrame
-
+from tabpfn_time_series.ts_dataframe import TimeSeriesDataFrame
 from tabpfn_time_series.tabpfn_worker import TabPFNClient, LocalTabPFN, MockTabPFN
 from tabpfn_time_series.defaults import TABPFN_TS_DEFAULT_CONFIG
 

tabpfn_time_series/tabpfn_worker.py

@@ -7,8 +7,8 @@ import pandas as pd
 import numpy as np
 import torch
 from scipy.stats import norm
-from autogluon.timeseries import TimeSeriesDataFrame
 
+from tabpfn_time_series.ts_dataframe import TimeSeriesDataFrame
 from tabpfn_time_series.data_preparation import split_time_series_to_X_y
 from tabpfn_time_series.defaults import TABPFN_TS_DEFAULT_QUANTILE_CONFIG