lint

Author: kleinhenz

src/e3tools/nn/__init__.py

@@ -10,24 +10,25 @@
 from ._scaling import ScaleIrreps
 
 __all__ = [
+    "Attention",
+    "AxisToMul",
     "Conv",
     "ConvBlock",
+    "EquivariantMLP",
     "ExperimentalConv",
-    "SeparableConv",
+    "ExperimentalTensorProduct",
+    "ExtractIrreps",
     "Gate",
-    "Gated",
     "GateWrapper",
-    "LinearSelfInteraction",
+    "Gated",
     "LayerNorm",
-    "EquivariantMLP",
-    "ScalarMLP",
-    "AxisToMul",
+    "LinearSelfInteraction",
     "MulToAxis",
-    "ExperimentalTensorProduct",
-    "SeparableTensorProduct",
-    "Attention",
     "MultiheadAttention",
-    "TransformerBlock",
-    "ExtractIrreps",
+    "ScalarMLP",
     "ScaleIrreps",
+    "SeparableConv",
+    "SeparableConvBlock",
+    "SeparableTensorProduct",
+    "TransformerBlock",
 ]