Skip to content

Commit ed768f0

Browse files
committed
chore: ruff
1 parent 00c161c commit ed768f0

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tests/app/services/invocation_stats/test_vram_tracking.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,8 +41,7 @@ def test_vram_tracking_with_no_gpu_operations():
4141

4242
# The peak VRAM should be 0 or very close to 0 since no GPU operations occurred
4343
assert node_stats.peak_vram_gb < 0.01, (
44-
f"Expected near-zero VRAM usage for node without GPU operations, "
45-
f"but got {node_stats.peak_vram_gb:.3f}G"
44+
f"Expected near-zero VRAM usage for node without GPU operations, but got {node_stats.peak_vram_gb:.3f}G"
4645
)
4746
finally:
4847
# Clean up
@@ -147,8 +146,7 @@ def test_vram_tracking_with_gpu_operations():
147146

148147
# The peak VRAM should reflect the allocation we made (roughly 400MB = 0.4GB)
149148
assert node_stats.peak_vram_gb > 0.1, (
150-
f"Expected significant VRAM usage for node with GPU operations, "
151-
f"but got {node_stats.peak_vram_gb:.3f}G"
149+
f"Expected significant VRAM usage for node with GPU operations, but got {node_stats.peak_vram_gb:.3f}G"
152150
)
153151
finally:
154152
# Clean up

0 commit comments

Comments
 (0)