Update CMakeLists.txt

open_spiel/python/CMakeLists.txt

@@ -266,13 +266,14 @@ set(PYTHON_TESTS ${PYTHON_TESTS}
 if (OPEN_SPIEL_ENABLE_JAX)
   # Only current JAX test is the bridge supervised learning example below.
   set (PYTHON_TESTS ${PYTHON_TESTS}
-    jax/deep_cfr_jax_test.py
-    jax/dqn_jax_test.py
-    jax/nfsp_jax_test.py
-    jax/opponent_shaping_jax_test.py
-    jax/policy_gradient_jax_test.py
-    algorithms/rnad/rnad_test.py
-    mfg/algorithms/fictitious_play_test.py
+    # Disable JAX tests temporarily
+    #jax/deep_cfr_jax_test.py
+    #jax/dqn_jax_test.py
+    #jax/nfsp_jax_test.py
+    #jax/opponent_shaping_jax_test.py
+    #jax/policy_gradient_jax_test.py
+    #algorithms/rnad/rnad_test.py
+    #mfg/algorithms/fictitious_play_test.py
   )
 endif()
 
@@ -367,14 +368,15 @@ endforeach(py_test_file)
 # Additional tests (running examples as tests)
 # We don't generate these automatically because we may want custom parameters.
 if (OPEN_SPIEL_ENABLE_JAX AND NOT OPEN_SPIEL_BUILDING_WHEEL)
-  add_test(NAME python_examples_bridge_supervised_learning
-           COMMAND ${Python3_EXECUTABLE}
-           ${CMAKE_CURRENT_SOURCE_DIR}/examples/bridge_supervised_learning.py
-           --iterations 10
-           --eval_every 5
-           --data_path ${CMAKE_CURRENT_SOURCE_DIR}/examples/data/bridge)
-  set_property(TEST python_examples_bridge_supervised_learning
-      PROPERTY ENVIRONMENT
-      PYTHONPATH=${CMAKE_CURRENT_BINARY_DIR}:${CMAKE_CURRENT_SOURCE_DIR}/../..;
-      TEST_SRCDIR=${CMAKE_CURRENT_SOURCE_DIR}/../..)
+  # Disable JAX tests temporarily
+  #add_test(NAME python_examples_bridge_supervised_learning
+  #         COMMAND ${Python3_EXECUTABLE}
+  #         ${CMAKE_CURRENT_SOURCE_DIR}/examples/bridge_supervised_learning.py
+  #         --iterations 10
+  #         --eval_every 5
+  #         --data_path ${CMAKE_CURRENT_SOURCE_DIR}/examples/data/bridge)
+  #set_property(TEST python_examples_bridge_supervised_learning
+  #    PROPERTY ENVIRONMENT
+  #    PYTHONPATH=${CMAKE_CURRENT_BINARY_DIR}:${CMAKE_CURRENT_SOURCE_DIR}/../..;
+  #    TEST_SRCDIR=${CMAKE_CURRENT_SOURCE_DIR}/../..)
 endif()