diff --git a/studio/backend/tests/test_gpu_arbiter.py b/studio/backend/tests/test_gpu_arbiter.py index 2e75b9129b..30010b11ca 100644 --- a/studio/backend/tests/test_gpu_arbiter.py +++ b/studio/backend/tests/test_gpu_arbiter.py @@ -345,7 +345,9 @@ def test_the_safetensors_load_yields_a_gpu_it_lost_while_loading(): encoding = "utf-8" ) load_impl = route_src[route_src.index("async def _load_model_impl") :] - unsloth_load = load_impl.index("success = await asyncio.to_thread(\n backend.load_model,") + unsloth_load = load_impl.index( + "success = await asyncio.to_thread(\n backend.load_model," + ) tail = load_impl[unsloth_load:] guard = tail.index("if current_owner() != CHAT:") assert "await asyncio.to_thread(backend.unload_model, config.identifier)" in tail[guard:] diff --git a/tests/studio/test_model_picker_contracts.py b/tests/studio/test_model_picker_contracts.py index 7d8f7a6b78..67064e7581 100644 --- a/tests/studio/test_model_picker_contracts.py +++ b/tests/studio/test_model_picker_contracts.py @@ -755,5 +755,11 @@ def test_video_gallery_fetches_clips_as_their_cards_come_into_view(): eager = list(re.finditer(r"page\.videos\.forEach\(\(video\) => void ensureSrc\(video\)\)", src)) assert eager, "the jsdom/old-webview fallback fetch is missing" for match in eager: - assert 'typeof IntersectionObserver === "undefined"' in src[max(0, match.start() - 260) : match.start()] - assert re.search(r"if \(!selected\) return;\s*\n\s*void \(async \(\) => \{\s*\n\s*await ensureSrc\(selected\);", src) + assert ( + 'typeof IntersectionObserver === "undefined"' + in src[max(0, match.start() - 260) : match.start()] + ) + assert re.search( + r"if \(!selected\) return;\s*\n\s*void \(async \(\) => \{\s*\n\s*await ensureSrc\(selected\);", + src, + )