fmt: ran formatter

This commit is contained in:
2024-08-26 03:34:50 -05:00
parent 5425ce1362
commit 7beca501bf

View File

@@ -267,14 +267,20 @@ class DirectMonteCarloRun:
seeds = seed_sequence.spawn(self.config.monte_carlo_cycles) seeds = seed_sequence.spawn(self.config.monte_carlo_cycles)
raw_pool_results = list(pool.imap_unordered( raw_pool_results = list(
pool.imap_unordered(
self._wrapped_single_run, self._wrapped_single_run,
[ [
(model_name_pair, seed, self.config.write_successes_to_file) (
model_name_pair,
seed,
self.config.write_successes_to_file,
)
for seed in seeds for seed in seeds
], ],
self.config.chunk_size, self.config.chunk_size,
)) )
)
pool_results = sum(result[0] for result in raw_pool_results) pool_results = sum(result[0] for result in raw_pool_results)
@@ -304,7 +310,9 @@ class DirectMonteCarloRun:
delimiter=",", delimiter=",",
) )
else: else:
_logger.debug("Instructed to write results, but none obtained") _logger.debug(
"Instructed to write results, but none obtained"
)
_logger.debug(f"Pool results: {pool_results}") _logger.debug(f"Pool results: {pool_results}")