Merge pull request #67 from ontime-re/48-benchmarking-fix-zero-few-shot

48 benchmarking fix zero and full shot
ontime-re · Nov 4, 2024 · 14eb867 · 14eb867
2 parents ec442e3 + 49c11f2
commit 14eb867
Show file tree

Hide file tree

Showing 9 changed files with 2,303 additions and 1,945 deletions.
diff --git a/docs/user_guide/1_module/3-benchmarking/3.0-benchmarking.ipynb b/docs/user_guide/1_module/3-benchmarking/3.0-benchmarking.ipynb
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,10 +8,10 @@ packages = [{include = "ontime", from = "src"}]
 
 [tool.poetry.dependencies]
 python = ">=3.10,<3.13"
-darts = "^0.25.0"
+darts = "^0.30.0"
 catboost = "^1.2"
 altair = "^5.1.0"
-skforecast = "^0.11.0"
+skforecast = "^0.12.0"
 torch = ">=2.0.0, !=2.0.1, !=2.1.0"
 vegafusion-python-embed = "^1.4.1"
 vegafusion = {extras = ["embed"], version = "^1.4.1"}

diff --git a/src/ontime/module/benchmarking/__init__.py b/src/ontime/module/benchmarking/__init__.py
@@ -0,0 +1,4 @@
+from .benchmark_metric import BenchmarkMetric
+from .benchmark_model import AbstractBenchmarkModel, BenchmarkMode
+from .benchmark_dataset import BenchmarkDataset
+from .benchmark import Benchmark