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

Convert formatting to use ruff instead of yapf + isort (#26247)

Browse Source 
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Harry Mellor
2025-10-05 15:06:22 +01:00
committed by GitHub
parent 17edd8a807
commit d6953beb91
1508 changed files with 115244 additions and 94146 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/vllm_test_utils/setup.py
View File

@@ -4,7 +4,7 @@
from setuptools import setup
setup(
name='vllm_test_utils',
version='0.1',
packages=['vllm_test_utils'],
name="vllm_test_utils",
version="0.1",
packages=["vllm_test_utils"],
)

4
tests/vllm_test_utils/vllm_test_utils/blame.py
View File

@@ -26,7 +26,7 @@ def blame(func: Callable) -> Generator[BlameResult, None, None]:
```python
with blame(lambda: some_condition()) as result:
# do something
if result.found:
print(result.trace_stack)
"""
@@ -34,7 +34,7 @@ def blame(func: Callable) -> Generator[BlameResult, None, None]:
def _trace_calls(frame, event, arg=None):
nonlocal result
if event in ['call', 'return']:
if event in ["call", "return"]:
# for every function call or return
try:
# Temporarily disable the trace function

27
tests/vllm_test_utils/vllm_test_utils/monitor.py
View File

@@ -19,8 +19,8 @@ class MonitoredValues(Generic[_T]):
@contextlib.contextmanager
def monitor(
measure_func: Callable[[],
_T]) -> Generator[MonitoredValues[_T], None, None]:
measure_func: Callable[[], _T],
) -> Generator[MonitoredValues[_T], None, None]:
"""
Trace the function calls to continuously monitor the change of
a value.
@@ -28,23 +28,23 @@ def monitor(
Usage:
```python
def measure_func():
... # measure the current value
... # measure the current value
return current_value
with monitor(measure_func) as monitored_values:
# do something
monitored_values.values # all changes of the values
monitored_values.trace_stacks # trace stacks of every change
monitored_values.values # all changes of the values
monitored_values.trace_stacks # trace stacks of every change
```
"""
monitored_values = MonitoredValues[_T]()
def _trace_calls(frame, event, arg=None):
nonlocal monitored_values
if event in ['line']:
if event in ["line"]:
# triggered by every line of Python code.
# only Python functions will trigger it,
# c/cpp functions will not trigger it.
@@ -53,11 +53,14 @@ def monitor(
sys.settrace(None)
# do a measurement
current_value = measure_func()
if len(monitored_values.values
) == 0 or current_value != monitored_values.values[-1]:
if (
len(monitored_values.values) == 0
or current_value != monitored_values.values[-1]
):
monitored_values.values.append(current_value)
monitored_values.trace_stacks.append("".join(
traceback.format_stack()))
monitored_values.trace_stacks.append(
"".join(traceback.format_stack())
)
# Re-enable the trace function
sys.settrace(_trace_calls)
except NameError: