Skip to content

Commit 771769d

Browse files
joke1196sonartech
authored andcommitted
SONARPY-3306 Fix pytorch stubs (#477)
GitOrigin-RevId: bcd05ea88d342a9fc63dce84684f208f652b8de5
1 parent ce4da1a commit 771769d

3 files changed

Lines changed: 17 additions & 29 deletions

File tree

python-frontend/src/main/resources/org/sonar/python/types/custom_protobuf/torch.protobuf

Lines changed: 12 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -54,14 +54,11 @@ pin_memory
5454
builtins.bool"builtins.bool *0
5555

5656
pin_memory
57-
builtins.bool"builtins.bool �
57+
builtins.bool"builtins.bool �
5858
empty torch.empty"
59-
torch.Tensor" torch.Tensor*�
60-
size�
61-
1Union[builtins.int,typing.Sequence[builtins.int]]
62-
builtins.int" builtins.intN
63-
typing.Sequence[builtins.int]
64-
builtins.int" builtins.int"typing.Sequence*N
59+
torch.Tensor" torch.Tensor*&
60+
size
61+
builtins.int" builtins.int*N
6562
dtypeA
6663
Union[torch.dtype,None]
6764
torch.dtype" torch.dtype
@@ -79,14 +76,11 @@ pin_memory
7976
builtins.bool"builtins.bool *0
8077

8178
pin_memory
82-
builtins.bool"builtins.bool �
79+
builtins.bool"builtins.bool �
8380
zeros torch.zeros"
84-
torch.Tensor" torch.Tensor*�
85-
size�
86-
1Union[builtins.int,typing.Sequence[builtins.int]]
87-
builtins.int" builtins.intN
88-
typing.Sequence[builtins.int]
89-
builtins.int" builtins.int"typing.Sequence*N
81+
torch.Tensor" torch.Tensor*&
82+
size
83+
builtins.int" builtins.int*N
9084
dtypeA
9185
Union[torch.dtype,None]
9286
torch.dtype" torch.dtype
@@ -104,15 +98,12 @@ pin_memory
10498
builtins.bool"builtins.bool *0
10599

106100
pin_memory
107-
builtins.bool"builtins.bool �
101+
builtins.bool"builtins.bool �
108102
ones
109103
torch.ones"
110-
torch.Tensor" torch.Tensor*�
111-
size�
112-
1Union[builtins.int,typing.Sequence[builtins.int]]
113-
builtins.int" builtins.intN
114-
typing.Sequence[builtins.int]
115-
builtins.int" builtins.int"typing.Sequence*N
104+
torch.Tensor" torch.Tensor*&
105+
size
106+
builtins.int" builtins.int*N
116107
dtypeA
117108
Union[torch.dtype,None]
118109
torch.dtype" torch.dtype
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
1-
6ee3fec0bc7c5f83c2283ef8ca181a002611c7a10f87d7df89b0c83a2d3b979d
2-
6618ce32b379cb645facaa0a42f9e418527b42ea15f7806030a0935d61868608
1+
67f9e602c1e0bb81578735419ab9b7ad284b716d2c26e0aad65ba5478148799f
2+
5d483afafd5d1120be5a493f93222d02362e1464f1ff0806dfd570a78ca9ed01

python-frontend/typeshed_serializer/resources/custom/torch/__init__.pyi

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,7 @@ def tensor(
3333

3434

3535
def empty(
36-
size: Union[int, Sequence[int]],
37-
*,
36+
*size: int,
3837
dtype: Optional[dtype] = ...,
3938
layout: Optional[layout] = ...,
4039
device: Optional[Union[str, device]] = ...,
@@ -43,8 +42,7 @@ def empty(
4342
) -> Tensor: ...
4443

4544
def zeros(
46-
size: Union[int, Sequence[int]],
47-
*,
45+
*size: int,
4846
dtype: Optional[dtype] = ...,
4947
layout: Optional[layout] = ...,
5048
device: Optional[Union[str, device]] = ...,
@@ -53,8 +51,7 @@ def zeros(
5351
) -> Tensor: ...
5452

5553
def ones(
56-
size: Union[int, Sequence[int]],
57-
*,
54+
*size: int,
5855
dtype: Optional[dtype] = ...,
5956
layout: Optional[layout] = ...,
6057
device: Optional[Union[str, device]] = ...,

0 commit comments

Comments
 (0)