8
8
class TestLocalScalarDenseConverter (DispatchTestCase ):
9
9
@parameterized .expand (
10
10
[
11
- (
12
- torch .randn ((5 , 10 , 5 ), dtype = torch .float32 ),
13
- ),
14
- (
15
- torch .randint (- 10 , 10 , (5 , 1 , 15 ), dtype = torch .int32 ),
16
- ),
17
- (
18
- torch .randn ((1 ), dtype = torch .float32 ),
19
- ),
20
- (
21
- (torch .tensor ([- 2.4 ])),
22
- ),
23
- (
24
- (torch .tensor ([5.5 , 3.5 , 3.6 ])),
25
- ),
26
- (
27
- (torch .tensor ([True ])),
28
- ),
11
+ (torch .randn ((5 , 10 , 5 ), dtype = torch .float32 ),),
12
+ (torch .randint (- 10 , 10 , (5 , 1 , 15 ), dtype = torch .int32 ),),
13
+ (torch .randn ((1 ), dtype = torch .float32 ),),
14
+ ((torch .tensor ([- 2.4 ])),),
15
+ ((torch .tensor ([5.5 , 3.5 , 3.6 ])),),
16
+ ((torch .tensor ([True ])),),
29
17
(
30
18
torch .tensor (
31
19
[
@@ -44,9 +32,7 @@ class TestLocalScalarDenseConverter(DispatchTestCase):
44
32
]
45
33
),
46
34
),
47
- (
48
- (torch .tensor ([float ("inf" )])),
49
- ),
35
+ ((torch .tensor ([float ("inf" )])),),
50
36
]
51
37
)
52
38
def test_local_scalar_dense (self , data ):
0 commit comments