Update ext.py

8e0bd4af · Matthias Fey · GitHub · d60df4b8 · 8e0bd4af
Commit 8e0bd4af authored Aug 07, 2018 by Matthias Fey Committed by GitHub Aug 07, 2018
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

torch_scatter/utils/ext.py torch_scatter/utils/ext.py +2 -2

No files found.
--- a/torch_scatter/utils/ext.py
+++ b/torch_scatter/utils/ext.py
@@ -6,5 +6,5 @@ if torch.cuda.is_available():
 def get_func(name, tensor):
-    scatter = scatter_cuda if tensor.is_cuda else scatter_cpu
+    module = scatter_cuda if tensor.is_cuda else scatter_cpu
-    return getattr(scatter, name)
+    return getattr(module, name)