increase tolerance for cross platform compat

Author: lazyoracle

tests/test_orthonormal.py

@@ -21,7 +21,9 @@ def test_get_orthonormal_basis(Phi):
 
     # Check if the basis vectors are orthogonal to each other
     assert jnp.allclose(
-        jnp.dot(basis_vectors, basis_vectors.T), jnp.eye(basis_vectors.shape[0])
+        jnp.dot(basis_vectors, basis_vectors.T),
+        jnp.eye(basis_vectors.shape[0]),
+        atol=1e-6,
     ), f"<A, A.T> should be I, got {jnp.dot(basis_vectors, basis_vectors.T)}"
 
     # Check if bases is orthogonal to Phi