Fix directed/undirected pre-process option in CitationFull (#8179)

Fixes #8171
pyg-team · Oct 11, 2023 · 6938d7f · 6938d7f
1 parent e9f62c0
commit 6938d7f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)