diff --git a/torch_geometric/datasets/citation_full.py b/torch_geometric/datasets/citation_full.py index 70c2d6f51b76..a1a305ce064b 100644 --- a/torch_geometric/datasets/citation_full.py +++ b/torch_geometric/datasets/citation_full.py @@ -98,7 +98,8 @@ def raw_file_names(self) -> str: @property def processed_file_names(self) -> str: - return 'data.pt' + suffix = 'undirected' if self.to_undirected else 'directed' + return f'data_{suffix}.pt' def download(self): download_url(self.url.format(self.name), self.raw_dir)