tensorflow tests: enable type promotion

jcmgray · Oct 23, 2024 · 1067d8c · 1067d8c
1 parent de0671f
commit 1067d8c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/test_tensor/test_optimizers.py b/tests/test_tensor/test_optimizers.py
@@ -16,6 +16,11 @@
 found_jax = importlib.util.find_spec("jax") is not None
 found_tensorflow = importlib.util.find_spec("tensorflow") is not None
 
+if found_tensorflow:
+    import tensorflow.experimental.numpy as tnp
+
+    tnp.experimental_enable_numpy_behavior()
+
 jax_case = pytest.param(
     "jax", marks=pytest.mark.skipif(not found_jax, reason="jax not installed")
 )