From 6938d7f784ee609eb5d0349d5599c9473acd8796 Mon Sep 17 00:00:00 2001 From: Matthias Fey Date: Wed, 11 Oct 2023 15:22:58 +0200 Subject: [PATCH] Fix directed/undirected pre-process option in `CitationFull` (#8179) Fixes #8171 --- torch_geometric/datasets/citation_full.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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)