cleaner

torch_scatter/__init__.py

-from .version import __version__  # noqa
-from .functions import *  # noqa
+from functions.add import scatter_add_, scatter_add
+from functions.sub import scatter_sub_, scatter_sub
+from functions.mul import scatter_mul_, scatter_mul
+from functions.div import scatter_div_, scatter_div
+from functions.mean import scatter_mean_, scatter_mean
+from functions.max import scatter_max_, scatter_max
+from functions.min import scatter_min_, scatter_min
+
+from .version import __version__
+
+__all__ = [
+    'scatter_add_', 'scatter_add', 'scatter_sub_', 'scatter_sub',
+    'scatter_mul_', 'scatter_mul', 'scatter_div_', 'scatter_div',
+    'scatter_mean_', 'scatter_mean', 'scatter_max_', 'scatter_max',
+    'scatter_min_', 'scatter_min', '__version__'
+]

torch_scatter/functions/__init__.py

-from .add import scatter_add_, scatter_add
-from .sub import scatter_sub_, scatter_sub
-from .mul import scatter_mul_, scatter_mul
-from .div import scatter_div_, scatter_div
-from .mean import scatter_mean_, scatter_mean
-from .max import scatter_max_, scatter_max
-from .min import scatter_min_, scatter_min
-
-__all__ = [
-    'scatter_add_', 'scatter_add', 'scatter_sub_', 'scatter_sub',
-    'scatter_mul_', 'scatter_mul', 'scatter_div_', 'scatter_div',
-    'scatter_mean_', 'scatter_mean', 'scatter_max_', 'scatter_max',
-    'scatter_min_', 'scatter_min'
-]