Skip to content

Commit

Permalink
Merge pull request #23 from LSSTDESC/22-hanging-nzdir-for-small-data
Browse files Browse the repository at this point in the history
The total number of chunks was not computed properly
  • Loading branch information
joselotl authored Aug 15, 2024
2 parents 85ff7c8 + 71a70a2 commit aca3baf
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/rail/estimation/algos/nz_dir.py
Original file line number Diff line number Diff line change
Expand Up @@ -156,7 +156,7 @@ def run(self):
for s, e, test_data in iterator:
print(f"Process {self.rank} running estimator on chunk {s} - {e}")
if first:
total_chunks = int(np.ceil(self._input_length/(e-s)))
total_chunks = int(np.ceil(self._input_length/self.config.chunk_size))
chunk_number = s//self.config.chunk_size
self._process_chunk(first, total_chunks, chunk_number, test_data, bootstrap_matrix)
first = False
Expand Down

0 comments on commit aca3baf

Please sign in to comment.