Skip to content

Commit 8c0d609

Browse files
committed
Remove PyTorch and Pyro-PPL from workflows
- Remove torch, torchvision, torchaudio installation from all workflows - Remove pyro-ppl installation from all workflows - Update step names from 'Install JAX, Numpyro, PyTorch' to 'Install JAX and Numpyro' - Disable build cache in ci.yml for full test run - Keep JAX and NumPyro installations
1 parent b61b30e commit 8c0d609

File tree

3 files changed

+14
-18
lines changed

3 files changed

+14
-18
lines changed

.github/workflows/cache.yml

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,12 +20,11 @@ jobs:
2020
python-version: "3.13"
2121
environment-file: environment.yml
2222
activate-environment: quantecon
23-
- name: Install JAX, Numpyro, PyTorch
23+
- name: Install JAX and Numpyro
2424
shell: bash -l {0}
2525
run: |
26-
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
2726
pip install --upgrade "jax[cuda12-local]==0.6.2"
28-
pip install numpyro
27+
pip install numpyro
2928
python scripts/test-jax-install.py
3029
- name: Check nvidia drivers
3130
shell: bash -l {0}

.github/workflows/ci.yml

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -28,13 +28,11 @@ jobs:
2828
python-version: "3.13"
2929
environment-file: environment.yml
3030
activate-environment: quantecon
31-
- name: Install JAX, Numpyro, PyTorch
31+
- name: Install JAX and Numpyro
3232
shell: bash -l {0}
3333
run: |
34-
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
35-
pip install pyro-ppl
36-
pip install --upgrade "jax[cuda12-local]==0.6.2"
37-
pip install numpyro pyro-ppl
34+
pip install --upgrade "jax[cuda12-local]"
35+
pip install numpyro
3836
python scripts/test-jax-install.py
3937
- name: Check nvidia Drivers
4038
shell: bash -l {0}
@@ -45,13 +43,13 @@ jobs:
4543
- name: Display Pip Versions
4644
shell: bash -l {0}
4745
run: pip list
48-
- name: Download "build" folder (cache)
49-
uses: dawidd6/action-download-artifact@v11
50-
with:
51-
workflow: cache.yml
52-
branch: main
53-
name: build-cache
54-
path: _build
46+
# - name: Download "build" folder (cache)
47+
# uses: dawidd6/action-download-artifact@v11
48+
# with:
49+
# workflow: cache.yml
50+
# branch: main
51+
# name: build-cache
52+
# path: _build
5553
# Build Assets (Download Notebooks and PDF via LaTeX)
5654
- name: Build Download Notebooks (sphinx-tojupyter)
5755
shell: bash -l {0}

.github/workflows/publish.yml

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,11 @@ jobs:
1919
python-version: "3.13"
2020
environment-file: environment.yml
2121
activate-environment: quantecon
22-
- name: Install JAX, Numpyro, PyTorch
22+
- name: Install JAX and Numpyro
2323
shell: bash -l {0}
2424
run: |
25-
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
2625
pip install --upgrade "jax[cuda12-local]==0.6.2"
27-
pip install numpyro
26+
pip install numpyro
2827
python scripts/test-jax-install.py
2928
- name: Check nvidia drivers
3029
shell: bash -l {0}

0 commit comments

Comments
 (0)