From 076bc7b6dbe43f02b4e87c360d4a7c0f2cd8891f Mon Sep 17 00:00:00 2001
From: Alon Kukliansky <alonkukl@gmail.com>
Date: Mon, 1 Jul 2024 15:29:48 -0700
Subject: [PATCH] Enabling 64bit to have good accuracy

---
 tests/test_single_leg_tensor.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/test_single_leg_tensor.py b/tests/test_single_leg_tensor.py
index ac6d801..80c4f7d 100644
--- a/tests/test_single_leg_tensor.py
+++ b/tests/test_single_leg_tensor.py
@@ -3,6 +3,7 @@
 from random import randint
 from random import sample
 
+import jax
 import jax.numpy as jnp
 import pytest
 from bqskit.ir.gates import CXGate
@@ -15,6 +16,9 @@
 from qfactorjax.unitarymatrixjax import UnitaryMatrixJax
 
 
+jax.config.update('jax_enable_x64', True)
+
+
 @pytest.mark.parametrize(
     'num_qubits, N',
     [(randint(2, 7), randint(3, 10)) for _ in range(6)],