move again_end default computation inside conditional

HighDiceRoller · Jun 10, 2024 · 7efc9a3 · 7efc9a3
1 parent fee6dbb
commit 7efc9a3
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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):