Commit df7a0ca3 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Fix bug in legacy metadata

parent 37c6edcb

deepchem/data/datasets.py

+2 −1
Original line number Diff line number Diff line
@@ -2136,7 +2136,8 @@ class DiskDataset(Dataset): 
              "Selecting from input shard %d/%d for selection output shard %d" %

              (shard_num + 1, n_shards, select_shard_num + 1))

          if self.legacy_metadata:

            shard_len = len(X)

            ids = self.get_shard_ids(shard_num)

            shard_len = len(ids)

          else:

            shard_X_shape, _, _, _ = self._get_shard_shape(shard_num)

            if len(shard_X_shape) > 0:

CRA Git | Maintained and supported by SUSTech CRA and CCSE