Skip to content

Commit ece31cc

Browse files
authored
Fix MoleculeGPTDataset paper link (#10457)
1 parent e061799 commit ece31cc

File tree

3 files changed

+7
-6
lines changed

3 files changed

+7
-6
lines changed

torch_geometric/data/dataset.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -236,8 +236,8 @@ def has_process(self) -> bool:
236236

237237
def _process(self):
238238
f = osp.join(self.processed_dir, 'pre_transform.pt')
239-
if osp.exists(f) and torch.load(f, weights_only=False) != _repr(
240-
self.pre_transform):
239+
if not self.force_reload and osp.exists(f) and torch.load(
240+
f, weights_only=False) != _repr(self.pre_transform):
241241
warnings.warn(
242242
"The `pre_transform` argument differs from the one used in "
243243
"the pre-processed version of this dataset. If you want to "
@@ -246,8 +246,8 @@ def _process(self):
246246
stacklevel=2)
247247

248248
f = osp.join(self.processed_dir, 'pre_filter.pt')
249-
if osp.exists(f) and torch.load(f, weights_only=False) != _repr(
250-
self.pre_filter):
249+
if not self.force_reload and osp.exists(f) and torch.load(
250+
f, weights_only=False) != _repr(self.pre_filter):
251251
warnings.warn(
252252
"The `pre_filter` argument differs from the one used in "
253253
"the pre-processed version of this dataset. If you want to "

torch_geometric/datasets/molecule_gpt_dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,7 @@ def extract_name(
174174
class MoleculeGPTDataset(InMemoryDataset):
175175
r"""The dataset from the `"MoleculeGPT: Instruction Following Large
176176
Language Models for Molecular Property Prediction"
177-
<https://ai4d3.github.io/papers/34.pdf>`_ paper.
177+
<https://ai4d3.github.io/2023/papers/34.pdf>`_ paper.
178178
179179
Args:
180180
root (str): Root directory where the dataset should be saved.

torch_geometric/nn/models/basic_gnn.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -415,7 +415,8 @@ class GCN(BasicGNN):
415415
(default: :obj:`None`)
416416
jk (str, optional): The Jumping Knowledge mode. If specified, the model
417417
will additionally apply a final linear transformation to transform
418-
node embeddings to the expected output feature dimensionality.
418+
node embeddings to the expected output feature dimensionality,
419+
while default will not.
419420
(:obj:`None`, :obj:`"last"`, :obj:`"cat"`, :obj:`"max"`,
420421
:obj:`"lstm"`). (default: :obj:`None`)
421422
**kwargs (optional): Additional arguments of

0 commit comments

Comments
 (0)