Update point.py

Liiam99 · Apr 12, 2024 · 9c17b1d · 9c17b1d
1 parent 6703ce8
commit 9c17b1d
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/transforms/point.py b/src/transforms/point.py
@@ -166,7 +166,12 @@ def _process(self, data):
 
             # C++ geometric features computation on CPU
             if self.k_step < 0:
-                f = pgeof.compute_features(xyz, nn, nn_ptr, self.k_min, verbose=False)
+                f = pgeof.compute_features(
+                    xyz, 
+                    nn, 
+                    nn_ptr, 
+                    self.k_min, 
+                    verbose=False)
             else:
                 f = pgeof.compute_features_optimal(
                     xyz,
@@ -175,8 +180,7 @@ def _process(self, data):
                     self.k_min,
                     self.k_step,
                     self.k_min_search,
-                    verbose=False,
-                )
+                    verbose=False)
             f = torch.from_numpy(f)
 
             # Keep only required features