Fix style and run pre-commit check

dottxt-ai · Dec 20, 2024 · 40f6ad1 · 40f6ad1
1 parent e9a573a
commit 40f6ad1
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/outlines/processors/base_logits_processor.py b/outlines/processors/base_logits_processor.py
@@ -112,9 +112,7 @@ def _to_torch(tensor_like: Array) -> torch.Tensor:
             # https://ml-explore.github.io/mlx/build/html/usage/numpy.html#pytorch
             if tensor_like.dtype == mx.bfloat16:
                 tensor_like = tensor_like.astype(mx.float32)
-            return torch.from_dlpack(
-                np.array(tensor_like, copy=False)
-            )
+            return torch.from_dlpack(np.array(tensor_like, copy=False))
 
         elif is_jax_array_type(type(tensor_like)):
             import jax