black

Author: WeiguoMa

tests/test_backends.py

@@ -1357,12 +1357,15 @@ def to_backend(z):
         try:
             if "torch" in name:
                 import torch
+
                 return torch.as_tensor(z)
             if "jax" in name:
                 import jax.numpy as jnp
+
                 return jnp.asarray(z)
             if "tf" in name or "tensorflow" in name:
                 import tensorflow as tf
+
                 return tf.convert_to_tensor(z)
         except Exception:
             pass
@@ -1390,4 +1393,4 @@ def to_backend(z):
     yf = to_backend([2.0, 2.0, 2.0])
     expectedf = np.floor_divide(np.array([7.9, 8.1, -3.5]), np.array([2.0, 2.0, 2.0]))
     out = tc.backend.floor_divide(xf, yf)
-    np.testing.assert_array_equal(np.array(out), expectedf)
+    np.testing.assert_array_equal(np.array(out), expectedf)