From b5d6c7924cb30a78c70100a8f7eddd284e0dab41 Mon Sep 17 00:00:00 2001 From: Bharath Ramsundar Date: Tue, 23 Jun 2020 22:18:29 -0700 Subject: [PATCH] fix --- deepchem/molnet/load_function/qm9_datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepchem/molnet/load_function/qm9_datasets.py b/deepchem/molnet/load_function/qm9_datasets.py index 71c5ca6846..299dd0f6f7 100644 --- a/deepchem/molnet/load_function/qm9_datasets.py +++ b/deepchem/molnet/load_function/qm9_datasets.py @@ -47,7 +47,7 @@ def load_qm9(featurizer='CoulombMatrix', loaded, all_dataset, transformers = deepchem.utils.save.load_dataset_from_disk( save_folder) if loaded: - return qm7_tasks, all_dataset, transformers + return qm9_tasks, all_dataset, transformers if featurizer in ['CoulombMatrix', 'BPSymmetryFunctionInput', 'MP', 'Raw']: dataset_file = os.path.join(data_dir, "gdb9.sdf")