fix CI testing environment

bytedance · Dec 10, 2024 · f0bb39b · f0bb39b
1 parent a3cab4b
commit f0bb39b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -13,16 +13,16 @@ jobs:
           path: netobs
       - uses: actions/checkout@v4
         with:
-          repository: AllanChain/deepsolid
-          ref: update-with-force
+          repository: bytedance/deepsolid
+          ref: b1abaf5212e1bfe1ce01257a81bc63b820efc4d4
           path: deepsolid
       - uses: actions/setup-python@v5
         with:
           python-version: '3.9'
       - name: Install
         run: |
-          pip install -e deepsolid scipy==1.8.0
           pip install 'jax[cpu]'==0.2.26 -f https://storage.googleapis.com/jax-releases/jax_releases.html
+          pip install -e deepsolid opt_einsum==3.3.0
           pip install -e 'netobs[dev]'
       - name: Test
         working-directory: netobs
@@ -36,14 +36,15 @@ jobs:
       - uses: actions/checkout@v4
         with:
           repository: google-deepmind/ferminet
+          ref: 9a1deec8222c1838c5aafb65830fc41372c48677
           path: ferminet
       - uses: actions/setup-python@v5
         with:
-          python-version: '3.10'
+          python-version: '3.11'
       - name: Install
         run: |
+          pip install jax==0.4.35
           pip install -e ferminet
-          pip install 'jax[cpu]'==0.4.24 -f https://storage.googleapis.com/jax-releases/jax_releases.html
           pip install -e 'netobs[dev]'
       - name: Test
         working-directory: netobs

diff --git a/tests/cli_test.py b/tests/cli_test.py
@@ -14,7 +14,7 @@
 
 import pytest
 
-from netobs.cli import make_cli
+from netobs.cli import cli
 
 try:
     from netobs.adaptors.ferminet_vmc import FERMINET_VERSION
@@ -29,7 +29,6 @@
     reason="FermiNet main is required to test it",
 )
 def test_cli():
-    cli = make_cli()
     cli(
         [
             "@ferminet_vmc",