fix: zero graph raise err

MDIL-SNU · Oct 21, 2024 · f55bd44 · f55bd44
1 parent 99c5631
commit f55bd44
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sevenn/train/graph_dataset.py b/sevenn/train/graph_dataset.py
@@ -155,6 +155,10 @@ def process(self):
                 data = self.pre_transform(data)
             processed_graph_list.append(data)
 
+        if len(processed_graph_list) == 0:
+            # Can not save at all if there is no graph (error in PyG), raise an error
+            raise ValueError('Zero graph found after filtering')
+
         # save graphs, handled by torch_geometrics
         self.save(processed_graph_list, self.processed_paths[0])