feat: add ai4ts as a dependency and update requirements files;

WenjieDu · Sep 5, 2024 · bdcbace · bdcbace
1 parent 65e059e
commit bdcbace
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 9 deletions.
diff --git a/pypots/gungnir/client.py b/pypots/gungnir/client.py
@@ -5,12 +5,9 @@
 # Created by Wenjie Du <[email protected]>
 # License: BSD-3-Clause
 
-from ..utils.logging import logger
-
+from ai4ts.client import TimeSeriesAI
 
-class TimeSeriesAI:
-    def __init__(self):
-        pass
+from ..utils.logging import logger
 
 
 class Gungnir(TimeSeriesAI):

diff --git a/requirements/conda_env.yml b/requirements/conda_env.yml
@@ -19,7 +19,11 @@ dependencies:
     - conda-forge::matplotlib
     - conda-forge::tensorboard
     - conda-forge::scikit-learn
-    - conda-forge::pygrinder >=0.6.2
+    - conda-forge::pygrinder >=0.6.4
     - conda-forge::tsdb >=0.6.1
-    - conda-forge::benchpots >=0.2.1
+    - conda-forge::benchpots >=0.3
     - pytorch::pytorch >=1.10.0
+
+    - pip:
+        # not published on conda-forge yet
+        - ai4ts
diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -10,5 +10,6 @@ tensorboard
 scikit-learn
 torch>=1.10.0
 tsdb>=0.6.1
-pygrinder>=0.6.2
-benchpots>=0.2.1
+pygrinder>=0.6.4
+benchpots>=0.3
+ai4ts