diff --git a/vllm/distributed/device_communicators/shm_broadcast.py b/vllm/distributed/device_communicators/shm_broadcast.py index b572d722..75d84c7a 100644 --- a/vllm/distributed/device_communicators/shm_broadcast.py +++ b/vllm/distributed/device_communicators/shm_broadcast.py @@ -110,7 +110,8 @@ class ShmRingBuffer: lambda *args, **kwargs: None): try: self.shared_memory = shared_memory.SharedMemory(name=name) - assert self.shared_memory.size == self.total_bytes_of_buffer # noqa + assert ( + self.shared_memory.size == self.total_bytes_of_buffer) except FileNotFoundError: # we might deserialize the object in a different node # in this case, this object is not used, @@ -344,8 +345,8 @@ class MessageQueue: time.sleep(RINGBUFFER_SLEEP_INTERVAL) # if we wait for a long time, we should warn the user - if time.monotonic( - ) - start_time > VLLM_RINGBUFFER_WARNING_INTERVAL * n_warning: # noqa + if (time.monotonic() - start_time > + VLLM_RINGBUFFER_WARNING_INTERVAL * n_warning): logger.warning( "No available block found in %s second. ", VLLM_RINGBUFFER_WARNING_INTERVAL) @@ -398,8 +399,8 @@ class MessageQueue: time.sleep(RINGBUFFER_SLEEP_INTERVAL) # if we wait for a long time, we should warn the user - if time.monotonic( - ) - start_time > VLLM_RINGBUFFER_WARNING_INTERVAL * n_warning: # noqa + if (time.monotonic() - start_time > + VLLM_RINGBUFFER_WARNING_INTERVAL * n_warning): logger.warning( "No available block found in %s second. ", VLLM_RINGBUFFER_WARNING_INTERVAL)