From 7efc9a3279d64e768bfed5f6e7875c6fc75da5a3 Mon Sep 17 00:00:00 2001 From: Albert Julius Liu Date: Mon, 10 Jun 2024 01:32:51 -0700 Subject: [PATCH] move `again_end` default computation inside conditional --- src/icepool/population/again.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/icepool/population/again.py b/src/icepool/population/again.py index b2223284..c952867b 100644 --- a/src/icepool/population/again.py +++ b/src/icepool/population/again.py @@ -308,10 +308,8 @@ def evaluate_agains_using_depth(outcomes: Sequence, times: Sequence[int], if again_depth < 0: raise ValueError('again_depth cannot be negative.') - not_again_die = compute_not_again_die(outcomes, times) - if again_end is None: - again_end = not_again_die.zero_outcome() + again_end = compute_not_again_die(outcomes, times).zero_outcome() def replace_again(outcome): if isinstance(outcome, AgainExpression):