Fix bug in legacy metadata

deepchem · Aug 19, 2020 · df7a0ca · df7a0ca
1 parent 37c6edc
commit df7a0ca
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deepchem/data/datasets.py b/deepchem/data/datasets.py
@@ -2136,7 +2136,8 @@ def generator():
               "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: