diff --git a/sevenn/train/graph_dataset.py b/sevenn/train/graph_dataset.py index ef64842..4f4d8ab 100644 --- a/sevenn/train/graph_dataset.py +++ b/sevenn/train/graph_dataset.py @@ -193,6 +193,8 @@ def __init__( def load(self, path: str, data_cls=Data) -> None: super().load(path, data_cls) + if len(self) == 0: + warnings.warn(f'No graphs found {self.processed_paths[0]}') self._load_meta() def _load_meta(self) -> None: @@ -377,6 +379,8 @@ def _read_graph_dataset( ) -> list[AtomGraphData]: meta_f = filename.replace('.pt', '.yaml') orig_cutoff = cutoff + if not os.path.exists(filename): + raise FileNotFoundError(f'No such file: {filename}') if not os.path.exists(meta_f): warnings.warn('No meta info found, beware of cutoff...') else: