Fix linting pyt-team#2

PatRyg99 · Jul 8, 2024 · 72a9fae · 72a9fae
1 parent c75e4b0
commit 72a9fae
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/modules/data/utils/utils.py b/modules/data/utils/utils.py
@@ -455,8 +455,8 @@ def load_pointcloud_dataset(cfg):
 
     if cfg.pos_to_x:
         return torch_geometric.data.Data(x=pos, pos=pos, num_nodes=num_points)
-    else:
-        return torch_geometric.data.Data(pos=pos, num_nodes=num_points)
+
+    return torch_geometric.data.Data(pos=pos, num_nodes=num_points)
 
 
 def load_manual_pointcloud(pos_to_x: bool = False):
@@ -485,8 +485,8 @@ def load_manual_pointcloud(pos_to_x: bool = False):
             pos=pos,
             num_nodes=len(pos),
         )
-    else:
-        return torch_geometric.data.Data(
-            pos=pos,
-            num_nodes=len(pos),
-        )
+
+    return torch_geometric.data.Data(
+        pos=pos,
+        num_nodes=len(pos),
+    )