Adjusted module exports

aidos-lab · Aug 23, 2023 · 82248dd · 82248dd
1 parent 733b83c
commit 82248dd
Showing 1 changed file with 14 additions and 12 deletions.
diff --git a/torch_topological/nn/__init__.py b/torch_topological/nn/__init__.py
@@ -13,20 +13,22 @@
 from .alpha_complex import AlphaComplex
 from .cubical_complex import CubicalComplex
 from .vietoris_rips_complex import VietorisRipsComplex
+
 from .weighted_euler_characteristic_transform import WeightedEulerCurve
 from .weighted_euler_characteristic_transform import EulerDistance
 
+
 __all__ = [
-    'AlphaComplex',
-    'CubicalComplex',
-    'PersistenceInformation',
-    'SignatureLoss',
-    'SummaryStatisticLoss',
-    'VietorisRipsComplex',
-    'WassersteinDistance',
-    'SlicedWassersteinDistance',
-    'SlicedWassersteinKernel',
-    'MultiScaleKernel',
-    'WeightedEulerCurve',
-    'EulerDistance'
+    "AlphaComplex",
+    "CubicalComplex",
+    "EulerDistance",
+    "MultiScaleKernel",
+    "PersistenceInformation",
+    "SignatureLoss",
+    "SlicedWassersteinDistance",
+    "SlicedWassersteinKernel",
+    "SummaryStatisticLoss",
+    "VietorisRipsComplex",
+    "WassersteinDistance",
+    "WeightedEulerCurve"
 ]