Skip to content

Commit 24ee5c0

Browse files
authored
Merge pull request #302 from corochann/relgat_fix_example
fix to relgat
2 parents d69dba8 + d397389 commit 24ee5c0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/qm9/predict_qm9.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def __call__(self, atoms, adjs):
5555
def parse_arguments():
5656
# Lists of supported preprocessing methods/models.
5757
method_list = ['nfp', 'ggnn', 'schnet', 'weavenet', 'rsgcn', 'relgcn',
58-
'gat']
58+
'relgat']
5959
label_names = ['A', 'B', 'C', 'mu', 'alpha', 'homo', 'lumo', 'gap', 'r2',
6060
'zpve', 'U0', 'U', 'H', 'G', 'Cv']
6161
scale_list = ['standardize', 'none']

0 commit comments

Comments
 (0)