Fix (#35)

awslabs · Jun 28, 2020 · be40342 · be40342
1 parent 889be12
commit be40342
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/CONTRIBUTORS.md b/CONTRIBUTORS.md
@@ -10,3 +10,4 @@ Contributors
 * [Chengqiang Lu](https://github.com/geekinglcq): Alchemy dataset; MPNN, MGCN and SchNet
 * [Jiajing Hu](https://github.com/jjhu94): Weave
 * [Zhaoqiang Chen](https://github.com/autodataming): Update RDkit channel for conda installation
+* [Yue Zhong](https://github.com/YueZhong-bio): HadamardLinkPredictor, link prediction for ogbl-ppa
diff --git a/examples/reaction_prediction/rexgen_direct/find_reaction_center_eval.py b/examples/reaction_prediction/rexgen_direct/find_reaction_center_eval.py
@@ -24,7 +24,7 @@ def main(args):
         test_set = USPTOCenter('test', num_processes=args['num_processes'], load=args['load'])
     else:
         test_set = WLNCenterDataset(raw_file_path=args['test_path'],
-                                    mol_graph_path='test.bin',
+                                    mol_graph_path=args['test_path'] + '.bin',
                                     num_processes=args['num_processes'],
                                     load=args['load'])
     test_loader = DataLoader(test_set, batch_size=args['batch_size'],