Skip to content

Commit f16e422

Browse files
committed
py 2.7 fix
1 parent 9697129 commit f16e422

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torch_scatter/utils/gen.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,14 +6,14 @@ def gen(src, index, dim=-1, out=None, dim_size=None, fill_value=0):
66

77
# Automatically expand index tensor to the right dimensions.
88
if index.dim() == 1:
9-
index_size = [*repeat(1, src.dim())]
9+
index_size = list(repeat(1, src.dim()))
1010
index_size[dim] = src.size(dim)
1111
index = index.view(index_size).expand_as(src)
1212

1313
# Generate output tensor if not given.
1414
if out is None:
1515
dim_size = index.max() + 1 if dim_size is None else dim_size
16-
out_size = [*src.size()]
16+
out_size = list(src.size())
1717
out_size[dim] = dim_size
1818
out = src.new_full(out_size, fill_value)
1919

0 commit comments

Comments
 (0)