[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
+140 -21
View File
@@ -635,26 +635,122 @@ class TestBuildTreeKernelEfficient(CustomTestCase):
# fmt: off
torch.tensor(
[
[29889, 29974, 29945, 29900, 29974, 29922, 29930, 29958,
29889, 29974, 29930, 29945, 29974, 29922, 29930, 29958],
[22550, 4136, 16492, 8439, 29871, 2, 3001, 13,
2, 13, 29906, 29946, 2, 13, 29871, 259],
[
29889,
29974,
29945,
29900,
29974,
29922,
29930,
29958,
29889,
29974,
29930,
29945,
29974,
29922,
29930,
29958,
],
[
22550,
4136,
16492,
8439,
29871,
2,
3001,
13,
2,
13,
29906,
29946,
2,
13,
29871,
259,
],
],
),
torch.tensor(
[
[29946, 29945, 29953, 29906, 29896, 29945, 29900, 29906,
29896, 29945, 29906, 29953, 29896, 29945, 29906, 29946],
[29871, 2, 29901, 29889, 29871, 2, 395, 259,
29901, 29871, 2, 29889, 3001, 1234, 7146, 2186],
[
29946,
29945,
29953,
29906,
29896,
29945,
29900,
29906,
29896,
29945,
29906,
29953,
29896,
29945,
29906,
29946,
],
[
29871,
2,
29901,
29889,
29871,
2,
395,
259,
29901,
29871,
2,
29889,
3001,
1234,
7146,
2186,
],
],
),
torch.tensor(
[
[29946, 29974, 29945, 29930, 29889, 29922, 29974, 29930,
29974, 29946, 29930, 29922, 29889, 29974, 29945, 29922],
[29941, 29906, 2, 29946, 29871, 450, 319, 14990,
29946, 29941, 2, 29906, 29871, 2, 3001, 13],
[
29946,
29974,
29945,
29930,
29889,
29922,
29974,
29930,
29974,
29946,
29930,
29922,
29889,
29974,
29945,
29922,
],
[
29941,
29906,
2,
29946,
29871,
450,
319,
14990,
29946,
29941,
2,
29906,
29871,
2,
3001,
13,
],
],
),
# fmt: on
@@ -772,15 +868,38 @@ class TestReconstructIndicesFromTreeMask(CustomTestCase):
tree_mask = torch.tensor(
# fmt: off
[
1, 0, 0, 0,
1, 1, 0, 0,
1, 0, 1, 0,
1, 0, 1, 1,
1, 0, 0, 0,
1, 1, 0, 0,
1, 1, 1, 0,
1, 1, 1, 1,
1,
0,
0,
0,
1,
1,
0,
0,
1,
0,
1,
0,
1,
0,
1,
1,
1,
0,
0,
0,
1,
1,
0,
0,
1,
1,
1,
0,
1,
1,
1,
1,
],
# fmt: on
dtype=torch.bool,