Skip to content

Commit 6938d7f

Browse files
authored
Fix directed/undirected pre-process option in CitationFull (#8179)
Fixes #8171
1 parent e9f62c0 commit 6938d7f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

Diff for: torch_geometric/datasets/citation_full.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,8 @@ def raw_file_names(self) -> str:
9898

9999
@property
100100
def processed_file_names(self) -> str:
101-
return 'data.pt'
101+
suffix = 'undirected' if self.to_undirected else 'directed'
102+
return f'data_{suffix}.pt'
102103

103104
def download(self):
104105
download_url(self.url.format(self.name), self.raw_dir)

0 commit comments

Comments
 (0)