No GitHub dependencies (Lightning-AI#1149)

wasifferoze · Mar 18, 2024 · eec1d95 · eec1d95
1 parent 4f5031a
commit eec1d95
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/.github/azure-gpu-test.yml b/.github/azure-gpu-test.yml
@@ -41,7 +41,7 @@ jobs:
       displayName: "Image info & NVIDIA"
 
     - script: |
-        pip install '.[all,test]'
+        pip install '.[all,test]' 'lm_eval @ git+https://github.com/EleutherAI/lm-evaluation-harness.git@115206dc89dad67b8b'
       displayName: 'Install dependencies'
 
     - bash: |

diff --git a/.github/workflows/cpu-tests.yml b/.github/workflows/cpu-tests.yml
@@ -61,7 +61,7 @@ jobs:
 
     - name: Install all dependencies
       run: |
-        uv pip install --system -e '.[all,test]'
+        uv pip install --system -e '.[all,test]' 'lm_eval @ git+https://github.com/EleutherAI/lm-evaluation-harness.git@115206dc89dad67b8b'
         uv pip list
 
     - name: Run tests

diff --git a/pyproject.toml b/pyproject.toml
@@ -10,9 +10,8 @@ license = { file = "LICENSE" }
 
 dependencies = [
     "torch>=2.2.0",
-    "lightning @ git+https://github.com/Lightning-AI/lightning@75553845c6bbcc305fbae38a46ef4e532e4ac85a",
-    # TODO: install from PyPI when https://github.com/omni-us/jsonargparse/pull/466 is released
-    "jsonargparse[signatures] @ git+https://github.com/omni-us/jsonargparse",
+    "lightning==2.3.0.dev20240318",
+    "jsonargparse[signatures]>=4.27.6",
 ]
 
 [project.urls]
@@ -43,7 +42,6 @@ all = [
     "pyarrow",                   # litgpt.data.prepare_starcoder.py
     "tensorboard",               # litgpt.pretrain
     "torchmetrics",              # litgpt.pretrain
-    "lm_eval @ git+https://github.com/EleutherAI/lm-evaluation-harness.git@115206dc89dad67b8b",
     "huggingface_hub[hf_transfer]>=0.21.0"  # download
 ]