[CI][RFC] Replace black-jupyter with ruff-format (#37210)

Co-authored-by: Alison Shao <a.shao@wustl.edu>
This commit is contained in:
Alex Nails
2026-09-02 19:46:08 -07:00
committed by GitHub
co-authored by Alison Shao
parent 2641e427be
commit 28262c20df
1411 changed files with 7766 additions and 8176 deletions
+6 -7
View File
@@ -29,7 +29,6 @@ TORCH_DTYPES = [torch.float16]
class TestClipModels(unittest.TestCase):
@classmethod
def setUpClass(cls):
mp.set_start_method("spawn", force=True)
@@ -61,12 +60,12 @@ class TestClipModels(unittest.TestCase):
)
print("text similarity diff", abs(text_similarity - 1))
print("image similarity diff", abs(image_similarity - 1))
assert torch.all(
abs(text_similarity - 1) < prefill_tolerance
), "embeddings are not all close"
assert torch.all(
abs(image_similarity - 1) < prefill_tolerance
), "embeddings are not all close"
assert torch.all(abs(text_similarity - 1) < prefill_tolerance), (
"embeddings are not all close"
)
assert torch.all(abs(image_similarity - 1) < prefill_tolerance), (
"embeddings are not all close"
)
def test_accuracy(self):
for model, prefill_tolerance in MODELS: