Merge branch 'main' of https://github.com/abearab/ezformer

abearab · Nov 22, 2024 · 9cad4bf · 9cad4bf
2 parents b0ee89c + fb8f599
commit 9cad4bf
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/.github/workflows/python-pkg.yml b/.github/workflows/python-pkg.yml
@@ -35,7 +35,7 @@ jobs:
           miniconda-version: "latest"
           auto-update-conda: true
           mamba-version: "*"
-          python-version: ${{ matrix.python-version }}
+          python-version: "3.11"
           channels: conda-forge,bioconda,pytorch,nvidia
           environment-file: environment.yml
     - name: "Install pytest"

diff --git a/environment.yml b/environment.yml
@@ -6,18 +6,19 @@ channels:
   - conda-forge
 dependencies:
   - python==3.11
-  - pytorch
-  - pytorch-cuda==12.1
-  - torchvision
-  - torchaudio
-  - torchmetrics
-  - cuda-toolkit
+  - pytorch::pytorch
+  - pytorch::pytorch-cuda
+  - pytorch::torchvision
+  - pytorch::torchaudio
+  - conda-forge::torchmetrics
+  - conda-forge::cuda-toolkit
+  - bioconda::basenji
+  - conda-forge::anndata
   - numpy
   - pandas
   - python-igraph
   - leidenalg
   - scikit-learn
-  - bioconda::basenji
   - matplotlib
   - dill
   - pynvml