fix typo error

jongyaoY · Sep 11, 2024 · d62e387 · d62e387
1 parent 714d3ab
commit d62e387
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/fignet/data_loader.py b/fignet/data_loader.py
@@ -69,7 +69,7 @@ def collate_fn(batch: List[Graph]):
             for edge_typ in graph.edge_sets.keys():
                 if (
                     edge_typ == EdgeType.MESH_MESH
-                    or edge_typ == EdgeType.MESH_MESH
+                    or edge_typ == EdgeType.FACE_FACE
                 ):
                     graph.edge_sets[edge_typ].index += m_node_offset
                 elif edge_typ == EdgeType.OBJ_MESH:
@@ -78,6 +78,8 @@ def collate_fn(batch: List[Graph]):
                 elif edge_typ == EdgeType.MESH_OBJ:
                     graph.edge_sets[edge_typ].index[0, :] += m_node_offset
                     graph.edge_sets[edge_typ].index[1, :] += o_node_offset
+                else:
+                    raise TypeError(f"Unknown edge type {edge_typ}")
                 # Concatenate
                 batch_graph.edge_sets[edge_typ].index = torch.cat(
                     [