Unverified Commit 9af0485d authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #964 from peastman/loop 

Fixed infinite loop
parents 07cf663f 54b178c5

deepchem/data/datasets.py

+2 −1
Original line number Diff line number Diff line
@@ -748,6 +748,7 @@ class DiskDataset(Dataset): 
        num_local_batches = math.ceil(n_shard_samples / shard_batch_size)



        if n_shard_samples == 0:

          cur_shard += 1

          continue

        if not deterministic:

          sample_perm = np.random.permutation(n_shard_samples)

CRA Git | Maintained and supported by SUSTech CRA and CCSE