File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed
tests/app/services/invocation_stats Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments