From b1174eb7f1b26343c1f3f88064314dab9919d45f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=92=B1=E6=98=B1=E5=86=B0?= <qyb@bytedance.com>
Date: Tue, 10 Dec 2024 09:37:58 +0800
Subject: [PATCH] fix CI testing environment

---
 .github/workflows/test.yml | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 74f54d6..dd01223 100644
--- 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
           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'
       - name: Install
         run: |
+          pip install jax==0.4.35 -f https://storage.googleapis.com/jax-releases/jax_releases.html
           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