fix(nyz): fix wandb requirements bug

opendilab · Dec 13, 2024 · 317e775 · 317e775
1 parent aa86aa7
commit 317e775
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 13 deletions.
diff --git a/.github/workflows/unit_test.yml b/.github/workflows/unit_test.yml
@@ -26,11 +26,6 @@ jobs:
           python -m pip install .
           python -m pip install ".[test,k8s]"
           python -m pip install transformers
-          if python --version | grep -q "Python 3.7"; then
-              python -m pip install wandb==0.16.4
-          else
-              echo "Python version is not 3.7, skipping wandb installation"
-          fi
           ./ding/scripts/install-k8s-tools.sh
           make unittest
       - name: Upload coverage to Codecov
@@ -60,10 +55,5 @@ jobs:
           python -m pip install .
           python -m pip install ".[test,k8s]"
           python -m pip install transformers
-          if python --version | grep -q "Python 3.7"; then
-              python -m pip install wandb==0.16.4
-          else
-              echo "Python version is not 3.7, skipping wandb installation"
-          fi
           ./ding/scripts/install-k8s-tools.sh
           make benchmark
diff --git a/setup.py b/setup.py
@@ -60,9 +60,9 @@
         'DI-toolkit>=0.1.0',
         'trueskill',
         'tensorboardX>=2.2',
-        'wandb',
+        'wandb<=0.19.0',
         'matplotlib',
-        'easydict==1.9',
+        'easydict>=1.9',
         'pyyaml',
         'enum_tools',
         'cloudpickle',
@@ -78,7 +78,7 @@
         'sniffio',  # parallel
         'redis',  # parallel
         'mpire>=2.3.5',  # parallel
-        'einops<=0.8.0',
+        'einops',
     ],
     extras_require={
         'test': [
@@ -189,6 +189,7 @@
         'Programming Language :: Python :: 3.7',
         'Programming Language :: Python :: 3.8',
         'Programming Language :: Python :: 3.9',
+        'Programming Language :: Python :: 3.10',
         'Topic :: Scientific/Engineering :: Artificial Intelligence',
     ],
 )