Commit 7ade211c authored by nitinprakash96's avatar nitinprakash96
Browse files

return entire dataset when split is None

parent 1dd7ec35

deepchem/molnet/load_function/delaney_datasets.py

+14 −11
Original line number Diff line number Diff line
@@ -53,6 +53,9 @@ def load_delaney(featurizer='ECFP', split='index', reload=True): 
  for transformer in transformers:

    dataset = transformer.transform(dataset)



  if split == None:

    return delaney_tasks, (dataset, None, None), transformers

  else:

    splitters = {

        'index': deepchem.splits.IndexSplitter(),

        'random': deepchem.splits.RandomSplitter(),

CRA Git | Maintained and supported by SUSTech CRA and CCSE