formatting

qiskit-community · Jun 16, 2023 · 2d6609f · 2d6609f
1 parent 9b4f195
commit 2d6609f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/qiskit_dynamics/dispatch/backends/jax.py b/qiskit_dynamics/dispatch/backends/jax.py
@@ -26,7 +26,11 @@
 
     if version.parse(jax.__version__) >= version.parse("0.4.4"):
         import os
-        if version.parse(jax.__version__) > version.parse("0.4.6") or os.environ.get('JAX_JIT_PJIT_API_MERGE', None) != '0':
+
+        if (
+            version.parse(jax.__version__) > version.parse("0.4.6")
+            or os.environ.get("JAX_JIT_PJIT_API_MERGE", None) != "0"
+        ):
             warnings.warn(
                 "The functionality in the perturbation module of Qiskit Dynamics requires a JAX "
                 "version <= 0.4.6, due to a bug in JAX versions > 0.4.6. For versions 0.4.4, "