Commit 745059ec authored by Joe Gomes's avatar Joe Gomes Committed by GitHub
Browse files

Bug fix in Dataset select function 

This PR fixes:
- The `select` function incorrectly initialized Dataset objects.  Ran into this bug during train/test/valid splitting
parent 36c6054d

deepchem/datasets/__init__.py

+1 −1
Original line number Diff line number Diff line
@@ -342,7 +342,7 @@ class Dataset(object): 
      os.makedirs(select_dir)

    if not len(indices):

      return Dataset(

          data_dir=select_dir, metadata_row=[], verbosity=self.verbosity)

          data_dir=select_dir, metadata_rows=[], verbosity=self.verbosity)

    indices = np.array(sorted(indices)).astype(int)

    count, indices_count = 0, 0

    metadata_rows = []

CRA Git | Maintained and supported by SUSTech CRA and CCSE