style

Signed-off-by: Staszek Pasko <[email protected]>

Author: p88h

vllm/v1/serial_utils.py

@@ -140,7 +140,7 @@ def _encode_tensor(
         # this creates a copy of the tensor
         obj = obj.contiguous() if not obj.is_contiguous() else obj
         #  view the tensor as a 1D array of bytes
-        arr = obj.view((obj.numel(),)).view(torch.uint8).numpy()
+        arr = obj.view((obj.numel(), )).view(torch.uint8).numpy()
         if obj.nbytes < self.size_threshold:
             data = msgpack.Ext(CUSTOM_TYPE_RAW_VIEW, arr.data)
         else:
@@ -228,7 +228,7 @@ def _decode_tensor(self, arr: Any) -> torch.Tensor:
         # the returned memory is non-writeable.
         buffer = self.aux_buffers[data] if isinstance(data, int) \
             else bytearray(data)
-        arr = np.ndarray(buffer=buffer, dtype=np.uint8, shape=(len(buffer),))
+        arr = np.ndarray(buffer=buffer, dtype=np.uint8, shape=(len(buffer), ))
         torch_dtype = getattr(torch, dtype)
         assert isinstance(torch_dtype, torch.dtype)
         return torch.from_numpy(arr).view(torch_dtype).view(shape)