fix default JAX_ENABLE_X64 env value

pola-rs · May 19, 2024 · 7bf53dd · 7bf53dd
1 parent 5319780
commit 7bf53dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py-polars/polars/dataframe/frame.py b/py-polars/polars/dataframe/frame.py
@@ -1767,7 +1767,7 @@ def to_jax(
             "for specific installation recommendations for the Jax package",
         )
         enabled_double_precision = jx.config.jax_enable_x64 or bool(
-            int(os.environ.get("JAX_ENABLE_X64", "1"))
+            int(os.environ.get("JAX_ENABLE_X64", "0"))
         )
         if dtype:
             frame = self.cast(dtype)