Merge pull request #1920 from deepchem/qm9

Fixes qm9 molnet type
deepchem · Jun 24, 2020 · 9096263 · 9096263
2 parents 52f3d33 + b5d6c79
commit 9096263
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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")