Fix issue with test script

slaclab · Nov 3, 2023 · 0cb7eb0 · 0cb7eb0
1 parent 7e1abbe
commit 0cb7eb0
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 8 deletions.
diff --git a/python3_rhel7_env.yml b/python3_rhel7_env.yml
@@ -86,8 +86,6 @@ dependencies:
   - pip:
     - Cheetah3
     - ipaddress
-    - jax==0.4.16
-    - jaxlib==0.4.16
     - matlab-wrapper
     - opencv-contrib-python
     - p4p  

diff --git a/rhel7_devel.yml b/rhel7_devel.yml
@@ -83,8 +83,6 @@ dependencies:
   - pip:
     - Cheetah3
     - ipaddress
-    - jax>=0.4.16
-    - jaxlib>=0.4.16
     - matlab-wrapper
     - opencv-contrib-python
     - p4p

diff --git a/test.sh b/test.sh
@@ -3,11 +3,9 @@
 # Temporary fix for a jax incompatibility, can be removed once that is resolved
 conda remove --force --offline --yes jax
 conda remove --force --offline --yes jaxlib
-pip install jax==0.4.16
-pip install jaxlib==0.4.19
-pip install jaxtyping==0.2.21
+pip install jax
 
 # All tests to be run to verify the environment built correctly and will work as expected at run time
 git clone https://github.com/ChristopherMayes/Xopt.git
-cd Xopt
+pushd Xopt
 pytest