Unverified Commit 90962639 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #1920 from deepchem/qm9 

Fixes qm9 molnet type
parents 52f3d33f b5d6c792

deepchem/molnet/load_function/qm9_datasets.py

+1 −1
Original line number Diff line number Diff line
@@ -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")

CRA Git | Maintained and supported by SUSTech CRA and CCSE