depen: add benchpots as a dependency;

WenjieDu · Jun 8, 2024 · 21271f0 · 21271f0
1 parent c515d42
commit 21271f0
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -11,5 +11,6 @@ matplotlib
 tensorboard
 scikit-learn
 torch>=1.10.0
-tsdb>=0.2
-pygrinder>=0.4
+tsdb>=0.4
+pygrinder>=0.6
+benchpots>=0.1
diff --git a/setup.cfg b/setup.cfg
@@ -35,8 +35,9 @@ basic =
  tensorboard
  scikit-learn
  torch>=1.10.0
- tsdb>=0.2
- pygrinder>=0.4
+ tsdb>=0.4
+ pygrinder>=0.6
+ benchpots>=0.1
 
 # dependencies that are optional, torch-geometric are only needed for model Raindrop
 # but its installation takes too much time

diff --git a/setup.py b/setup.py
@@ -56,8 +56,9 @@
  "tensorboard",
  "scikit-learn",
  "torch>=1.10.0",
- "tsdb>=0.2",
- "pygrinder>=0.4",
+ "tsdb>=0.4",
+ "pygrinder>=0.6",
+ "benchpots>=0.1",
  ],
  python_requires=">=3.7.0",
  setup_requires=["setuptools>=38.6.0"],