[CI][RFC] Replace black-jupyter with ruff-format (#37210)
Co-authored-by: Alison Shao <a.shao@wustl.edu>
This commit is contained in:
co-authored by
Alison Shao
parent
2641e427be
commit
28262c20df
@@ -40,9 +40,9 @@ def test_stage_entry_sync_excludes_previous_stage_tail(monkeypatch):
|
||||
torch.ones(8, device="cuda").sum().cpu()
|
||||
|
||||
producer_ms, consumer_ms = metrics.stages["producer"], metrics.stages["consumer"]
|
||||
assert (
|
||||
producer_ms > 250
|
||||
), f"queued work not attributed to producer: {metrics.stages}"
|
||||
assert producer_ms > 250, (
|
||||
f"queued work not attributed to producer: {metrics.stages}"
|
||||
)
|
||||
assert consumer_ms < 100, f"producer tail leaked into consumer: {metrics.stages}"
|
||||
|
||||
|
||||
|
||||
@@ -26,7 +26,6 @@ register_cuda_ci(
|
||||
|
||||
|
||||
class TestStartProfile(CustomTestCase):
|
||||
|
||||
@classmethod
|
||||
def setUpClass(cls):
|
||||
cls.output_dir = tempfile.mkdtemp()
|
||||
|
||||
@@ -45,7 +45,6 @@ def _is_nsys_available():
|
||||
|
||||
|
||||
class TestStartProfile(CustomTestCase):
|
||||
|
||||
@classmethod
|
||||
def setUpClass(cls):
|
||||
envs.SGLANG_TORCH_PROFILER_DIR.set(OUTPUT_DIR)
|
||||
|
||||
Reference in New Issue
Block a user