diff --git a/src/elo.rs b/src/elo.rs index 0a07d3d..2c1af20 100644 --- a/src/elo.rs +++ b/src/elo.rs @@ -46,7 +46,7 @@ pub fn run() { } // different values of top_k_children - [1, 2, 3] + [1, 2] .map(move |top_k_children| FutureMoveConfig { top_k_children, ..prev_c @@ -56,12 +56,12 @@ pub fn run() { .flat_map(move |prev_c| { if !prev_c.do_prune { // do not bother making configs when pruning is disabled - // as top_k_children does nothing when pruning is skipped return vec![prev_c]; } - // different values of top_k_children - [1, 2, 3] + // different values to be subtracted from max_depth + // to become min_arena_depth + [prev_c.max_depth, 1, 3] .into_iter() .filter(|&x| x <= prev_c.max_depth) .map(move |ad_offset| FutureMoveConfig { @@ -69,6 +69,22 @@ pub fn run() { ..prev_c }) .collect() + }) + .flat_map(move |prev_c| { + if !prev_c.do_prune { + // do not bother making configs when pruning is disabled + return vec![prev_c]; + } + + // different values of up_to_minus + [0, 2, 4] + .into_iter() + .filter(|&x| x <= prev_c.max_depth) + .map(move |up_to_minus| FutureMoveConfig { + up_to_minus, + ..prev_c + }) + .collect() }); let vec: Vec<(String, Box Box>)> = configs