[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
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user