Update ext.py

Author: rusty1s

torch_scatter/utils/ext.py

@@ -6,5 +6,5 @@
 
 
 def get_func(name, tensor):
-    scatter = scatter_cuda if tensor.is_cuda else scatter_cpu
-    return getattr(scatter, name)
+    module = scatter_cuda if tensor.is_cuda else scatter_cpu
+    return getattr(module, name)