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

Merge pull request #2195 from deepchem/qm7_fix

Fixing QM7 loader
parents 6d480814 e9d35a2d
Loading
Loading
Loading
Loading
+1 −5
Original line number Diff line number Diff line
@@ -278,11 +278,7 @@ def load_qm7(featurizer='CoulombMatrix',
  qm7_tasks = ["u0_atom"]
  if featurizer == 'CoulombMatrix':
    featurizer = deepchem.feat.CoulombMatrixEig(23)
  loader = deepchem.data.SDFLoader(
      tasks=qm7_tasks,
      smiles_field="smiles",
      mol_field="mol",
      featurizer=featurizer)
  loader = deepchem.data.SDFLoader(tasks=qm7_tasks, featurizer=featurizer)
  dataset = loader.featurize(dataset_file)

  if split == None: