Skip to content

Commit 239647e

Browse files
author
Zhijian Liu
committed
[minor] add type annotation for KernelRegion
1 parent a1e478c commit 239647e

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

torchsparse/utils/kernel_region.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,10 @@
1212

1313
class KernelRegion:
1414
def __init__(self,
15-
kernel_size=3,
16-
tensor_stride=1,
17-
dilation=1,
18-
dim=[0, 1, 2]):
19-
15+
kernel_size: int = 3,
16+
tensor_stride: int = 1,
17+
dilation: int = 1,
18+
dim=[0, 1, 2]) -> None:
2019
self.kernel_size = kernel_size
2120
self.tensor_stride = tensor_stride
2221
self.dilation = dilation

0 commit comments

Comments
 (0)