Black reformatted to 80 chars

aidos-lab · Jul 4, 2024 · 871e5ba · 871e5ba
1 parent 046b26a
commit 871e5ba
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/dect/ect.py b/dect/ect.py
@@ -107,7 +107,9 @@ def compute_ecc(
     return segment_add_coo(ecc, index)
 
 
-def compute_ect_points(batch: Batch, v: torch.FloatTensor, lin: torch.FloatTensor):
+def compute_ect_points(
+    batch: Batch, v: torch.FloatTensor, lin: torch.FloatTensor
+):
     """Computes the Euler Characteristic Transform of a batch of point clouds.
 
     Parameters
@@ -127,7 +129,9 @@ def compute_ect_points(batch: Batch, v: torch.FloatTensor, lin: torch.FloatTenso
     return compute_ecc(nh, batch.batch, lin)
 
 
-def compute_ect_edges(data: Batch, v: torch.FloatTensor, lin: torch.FloatTensor):
+def compute_ect_edges(
+    data: Batch, v: torch.FloatTensor, lin: torch.FloatTensor
+):
     """Computes the Euler Characteristic Transform of a batch of graphs.
 
     Parameters
@@ -163,7 +167,9 @@ def compute_ect_edges(data: Batch, v: torch.FloatTensor, lin: torch.FloatTensor)
     )
 
 
-def compute_ect_faces(data: Batch, v: torch.FloatTensor, lin: torch.FloatTensor):
+def compute_ect_faces(
+    data: Batch, v: torch.FloatTensor, lin: torch.FloatTensor
+):
     """Computes the Euler Characteristic Transform of a batch of meshes.
 
     Parameters
@@ -245,9 +251,9 @@ def __init__(self, config: ECTConfig, v=None):
         super().__init__()
         self.config = config
         self.lin = nn.Parameter(
-            torch.linspace(-config.radius, config.radius, config.bump_steps).view(
-                -1, 1, 1, 1
-            ),
+            torch.linspace(
+                -config.radius, config.radius, config.bump_steps
+            ).view(-1, 1, 1, 1),
             requires_grad=False,
         )