fixup! Avoid JIT compilation errors

nengo · Jul 11, 2024 · 55bb094 · 55bb094
1 parent 38bc180
commit 55bb094
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/keras_lmu/tests/conftest.py b/keras_lmu/tests/conftest.py
@@ -7,5 +7,7 @@
 
 def pytest_configure(config):
     tf.debugging.disable_traceback_filtering()
+    tf.config.experimental.enable_op_determinism()
+    tf.keras.utils.set_random_seed(0)
     if version.parse(tf.__version__) >= version.parse("2.16.0"):
         keras.config.disable_traceback_filtering()