squall/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[Misc] Remove abused noqa (#6619)

Browse Source
This commit is contained in:
Woosuk Kwon 2024-07-21 08:47:04 -07:00 committed by GitHub
parent 14f91fe67c
commit b6df37f943
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
vllm/distributed/device_communicators/shm_broadcast.py
View File

@ -110,7 +110,8 @@ class ShmRingBuffer:
lambda *args, **kwargs: None): lambda *args, **kwargs: None):
try: try:
self.shared_memory = shared_memory.SharedMemory(name=name) 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: except FileNotFoundError:
# we might deserialize the object in a different node # we might deserialize the object in a different node
# in this case, this object is not used, # in this case, this object is not used,
@ -344,8 +345,8 @@ class MessageQueue:
time.sleep(RINGBUFFER_SLEEP_INTERVAL) time.sleep(RINGBUFFER_SLEEP_INTERVAL)
# if we wait for a long time, we should warn the user # if we wait for a long time, we should warn the user
if time.monotonic( if (time.monotonic() - start_time >
) - start_time > VLLM_RINGBUFFER_WARNING_INTERVAL * n_warning: # noqa VLLM_RINGBUFFER_WARNING_INTERVAL * n_warning):
logger.warning( logger.warning(
"No available block found in %s second. ", "No available block found in %s second. ",
VLLM_RINGBUFFER_WARNING_INTERVAL) VLLM_RINGBUFFER_WARNING_INTERVAL)
@ -398,8 +399,8 @@ class MessageQueue:
time.sleep(RINGBUFFER_SLEEP_INTERVAL) time.sleep(RINGBUFFER_SLEEP_INTERVAL)
# if we wait for a long time, we should warn the user # if we wait for a long time, we should warn the user
if time.monotonic( if (time.monotonic() - start_time >
) - start_time > VLLM_RINGBUFFER_WARNING_INTERVAL * n_warning: # noqa VLLM_RINGBUFFER_WARNING_INTERVAL * n_warning):
logger.warning( logger.warning(
"No available block found in %s second. ", "No available block found in %s second. ",
VLLM_RINGBUFFER_WARNING_INTERVAL) VLLM_RINGBUFFER_WARNING_INTERVAL)