fix black

Author: Stellogic

examples/lennard_jones_optimization.py

@@ -14,15 +14,13 @@
 import matplotlib.pyplot as plt
 
 # Try to enable JAX 64-bit precision if available (safe fallback)
-import jax  # noqa: E402
 
 try:  # pragma: no cover - optional optimization
     from jax import config as jax_config  # type: ignore
 
     jax_config.update("jax_enable_x64", True)
 except Exception:  # broad: environment may not have config attribute
     pass
-import jax.numpy as jnp  # noqa: E402
 import tensorcircuit as tc  # noqa: E402
 
 

tests/test_backends.py

@@ -1194,12 +1194,12 @@ def test_backend_all(backend):
     # Test all True
     a = tc.backend.convert_to_tensor([True, True, True])
     result = tc.backend.all(a)
-    assert result == True
+    assert result is True
 
     # Test with False
     b = tc.backend.convert_to_tensor([True, False, True])
     result = tc.backend.all(b)
-    assert result == False
+    assert result is False
 
 
 @pytest.mark.parametrize("backend", [lf("npb"), lf("tfb"), lf("jaxb"), lf("torchb")])