minor change

import shutil

import deepchem as dc

def load_sampl(featurizer='ECFP', split='random', frac_train=0.8):

def load_sampl(featurizer='ECFP', split='index'):

  """Load SAMPL datasets."""

  # Featurize SAMPL dataset

  print("About to featurize SAMPL dataset.")

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

               'scaffold': dc.splits.ScaffoldSplitter()}

  splitter = splitters[split]

  train, valid, test = splitter.train_valid_test_split(dataset, frac_train=frac_train,

                             frac_valid=1-frac_train, frac_test=0.)

  train, valid, test = splitter.train_valid_test_split(dataset)

  return SAMPL_tasks, (train, valid, test), transformers