AdityaLab · kage08 · Mar 26, 2025 · Feb 21, 2025 · Feb 21, 2025 · Feb 21, 2025
diff --git a/example/lptm.ipynb b/example/lptm.ipynb
diff --git a/example/lptm_zero.ipynb b/example/lptm_zero.ipynb
diff --git a/example/moirai.ipynb b/example/moirai.ipynb
diff --git a/example/timesfm.ipynb b/example/timesfm.ipynb
diff --git a/example/tinytimemixer.ipynb b/example/tinytimemixer.ipynb
diff --git a/leaderboard.py b/leaderboard.py
@@ -8,10 +8,10 @@
 # if src_path not in sys.path:
 #     sys.path.insert(0, src_path)
 
-from src.samay.model import TimesfmModel, MomentModel, ChronosModel, ChronosBoltModel, TinyTimeMixerModel
-from src.samay.dataset import TimesfmDataset, MomentDataset, ChronosDataset, ChronosBoltDataset, TinyTimeMixerDataset
-from src.samay.utils import load_args
-from src.samay.metric import *
+from samay.model import TimesfmModel, MomentModel, ChronosModel, ChronosBoltModel, TinyTimeMixerModel
+from samay.dataset import TimesfmDataset, MomentDataset, ChronosDataset, ChronosBoltDataset, TinyTimeMixerDataset
+from samay.utils import load_args
+from samay.metric import *
 
 
 ECON_NAMES = {

diff --git a/pyproject.toml b/pyproject.toml
@@ -23,11 +23,12 @@ dependencies = [
     "datasets>=3.2.0",
     "chronos-forecasting>=1.4.1",
     "tensorboardx>=2.6.2.2",
+    "einops>=0.8.1",
 ]
 
 [build-system]
-requires = ["maturin>=1.7,<2.0"]
-build-backend = "maturin"
+requires = ["hatchling"]
+build-backend = "hatchling.build"
 
 [dependency-groups]
 dev = [
@@ -37,7 +38,3 @@ dev = [
     "ruff>=0.8.1",
     "wandb>=0.18.5",
 ]
-
-[tool.maturin]
-features = ["pyo3/extension-module"]
-module-name = "samay.models.lptm.selection"
diff --git a/rust/.gitignore b/rust/.gitignore
diff --git a/rust/Cargo.toml b/rust/Cargo.toml
diff --git a/rust/src/lib.rs b/rust/src/lib.rs