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

Merge pull request #1540 from ihaque/fix-muv-loading

Fix loading of MUV with `reload = False`
parents 6c1ded50 0a6d8ee3
Loading
Loading
Loading
Loading
+3 −3
Original line number Diff line number Diff line
@@ -76,8 +76,8 @@ def load_muv(featurizer='ECFP', split='index', reload=True, K=4):
    all_dataset = fold_datasets
  else:
    train, valid, test = splitter.train_valid_test_split(dataset)
    all_dataset = (train, valid, test)
    if reload:
      deepchem.utils.save.save_dataset_to_disk(save_dir, train, valid, test,
                                               transformers)
    all_dataset = (train, valid, test)
  return MUV_tasks, all_dataset, transformers