Updating to get rid of warnings.

leakec · Dec 5, 2023 · 9d30670 · 9d30670
1 parent 39865ab
commit 9d30670
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 6 deletions.
diff --git a/tests/test_BF.py b/tests/test_BF.py
@@ -1,6 +1,3 @@
-from jax.config import config
-
-config.update("jax_enable_x64", True)
 import jax.numpy as np
 
 from tfc.utils.BF import (

diff --git a/tests/test_nBF.py b/tests/test_nBF.py
@@ -1,6 +1,3 @@
-from jax.config import config
-
-config.update("jax_enable_x64", True)
 import numpy as onp
 import jax.numpy as np