diff --git a/torch_geometric/datasets/git_mol_dataset.py b/torch_geometric/datasets/git_mol_dataset.py index ef6ac0a13a4e..a6fc67f63d54 100644 --- a/torch_geometric/datasets/git_mol_dataset.py +++ b/torch_geometric/datasets/git_mol_dataset.py @@ -206,7 +206,7 @@ def process(self) -> None: edges_list = [] edge_features_list = [] - for bond in mol.GetBonds(): + for bond in mol.GetBonds(): # type: ignore i, j = bond.GetBeginAtomIdx(), bond.GetEndAtomIdx() edge_feature = [ safe_index(