Skip to content

Commit b42c862

Browse files
committed
fix
1 parent f89731c commit b42c862

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

ppmat/models/newtonnet/scatter.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from paddle_geometric.utils import scatter as scatter_org
22

3+
# The parameter may need to be converted to an integer before calling the original function, so add the function here
34
def scatter(src, index, dim=0, dim_size=None, reduce='sum'):
45
return scatter_org(src, index, dim=dim, dim_size=dim_size, reduce=reduce)
56

0 commit comments

Comments
 (0)