From 7bf53ddf86b86afd4b5e548e160e47c285700026 Mon Sep 17 00:00:00 2001 From: alexander-beedie Date: Sat, 18 May 2024 16:06:12 +0400 Subject: [PATCH] fix default `JAX_ENABLE_X64` env value --- py-polars/polars/dataframe/frame.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py-polars/polars/dataframe/frame.py b/py-polars/polars/dataframe/frame.py index ecd9552f1d0a..257d806b258c 100644 --- 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)