Update ci pipeline

.github/workflows/build.yml

@@ -6,7 +6,7 @@ on:
 
 jobs:
   build-gpu-image:
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-24.04
     steps:
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3
@@ -29,7 +29,7 @@ jobs:
           ENV: prod
           PLATFORM: gpu
   build-cpu-linux-amd64:
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-24.04
     steps:
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3
@@ -52,7 +52,7 @@ jobs:
           ENV: prod
           PLATFORM: cpu
   build-cpu-linux-arm64:
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-24.04
     steps:
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3

.github/workflows/lint.yml

@@ -4,7 +4,7 @@ on: [push]
 
 jobs:
   ruff:
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-24.04
     strategy:
       matrix:
         python-version: ["3.9"]

.github/workflows/pytest.yml

@@ -11,7 +11,7 @@ on:
 
 jobs:
   pytest-unit:
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-24.04
     strategy:
       matrix:
         python-version: ["3.9"]
@@ -57,7 +57,7 @@ jobs:
         env:
           CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
   pytest-integrative-1:
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-24.04
     strategy:
       matrix:
         python-version: ["3.9"]
@@ -103,7 +103,7 @@ jobs:
         env:
           CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
   pytest-integrative-2:
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-24.04
     strategy:
       matrix:
         python-version: ["3.9"]
@@ -149,7 +149,7 @@ jobs:
         env:
           CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
   pytest-integrative-3:
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-24.04
     strategy:
       matrix:
         python-version: ["3.9"]
@@ -195,7 +195,7 @@ jobs:
         env:
           CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
   pytest-integrative-4:
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-24.04
     strategy:
       matrix:
         python-version: ["3.9"]

pip/cpu_requirements.txt

@@ -1,7 +1,7 @@
 # Minimal pip requirement list for AMPL with CPU support only. This includes only the packages directly imported by AMPL code, plus a few others commonly used during development.
 -i https://download.pytorch.org/whl/cpu
---extra-index-url https://data.dgl.ai/wheels/repo.html
---extra-index-url https://pypi.python.org/simple
+--find-links https://data.dgl.ai/wheels/repo.html
+--extra-index-url  https://pypi.python.org/simple
 # for LC developers: comment out pypi index url and use t$
 # --extra-index-url https://wci-repo.llnl.gov/repository/$
 

pip/cuda_requirements.txt

@@ -1,6 +1,6 @@
 # Minimal pip requirement list for AMPL with CUDA-enabled GPUs. This includes only the packages directly imported by AMPL code, plus a few others commonly used during development. It requires CUDA-enable GPUs and nvidia packages to be installed first.
 -i https://download.pytorch.org/whl/cu118
---extra-index-url https://pypi.python.org/simple
+--extra-index-url  https://pypi.python.org/simple
 # **Note**: for LC developers, comment out line 3 pypi install and use/uncommenting line 5:
 #--extra-index-url https://wci-repo.llnl.gov/repository/pypi-group/simple
 -f https://data.dgl.ai/wheels/cu118/repo.html

pip/docker_requirements.txt

@@ -1,7 +1,7 @@
 # Minimal pip requirement list for AMPL with CPU support only. This includes only the packages directly imported by AMPL code, plus a few others commonly used during development.
 -i https://download.pytorch.org/whl/cpu
---extra-index-url https://data.dgl.ai/wheels/repo.html
---extra-index-url https://pypi.python.org/simple
+--find-links https://data.dgl.ai/wheels/repo.html
+--extra-index-url  https://pypi.python.org/simple
 # for LC developers: comment out pypi index url and use t$
 # --extra-index-url https://wci-repo.llnl.gov/repository/$
 

pip/mchip_requirements.txt

@@ -1,7 +1,7 @@
 # Minimal pip requirement list for AMPL with CPU support only for MacOS ARM chip. This includes only the packages directly imported by AMPL code, plus a few others commonly used during development. 
 -i https://download.pytorch.org/whl/cpu
---extra-index-url https://data.dgl.ai/wheels/repo.html
---extra-index-url https://pypi.python.org/simple
+--find-links https://data.dgl.ai/wheels/repo.html
+--extra-index-url  https://pypi.python.org/simple
 # for LC developers: comment out pypi index url and use t$
 # --extra-index-url https://wci-repo.llnl.gov/repository/$