From 2f87e2aa60fa5b3de0a99309f4377bf24997aa52 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Max=20K=C3=A4nner?= Date: Fri, 6 Jun 2025 21:38:34 +0200 Subject: [PATCH] improve log output --- battlesnake/src/main.rs | 19 ++++++------------- 1 file changed, 6 insertions(+), 13 deletions(-) diff --git a/battlesnake/src/main.rs b/battlesnake/src/main.rs index 02acff8..ff4750e 100644 --- a/battlesnake/src/main.rs +++ b/battlesnake/src/main.rs @@ -124,7 +124,6 @@ async fn get_move(request: Json) -> response::Json { let request = request.clone(); let board = board.clone(); let mut rng = SmallRng::from_os_rng(); - let actions = actions.clone(); tokio::task::spawn_blocking(move || { let mut mcts_managers: Vec<_> = (0..request.board.snakes.len()) .map(|id| MctsManager::new(u8::try_from(id).unwrap())) @@ -153,16 +152,6 @@ async fn get_move(request: Json) -> response::Json { } } let my_mcts_manager = mcts_managers.into_iter().nth(usize::from(id)).unwrap(); - for action in actions { - let score = my_mcts_manager.base.next[usize::from(action)] - .as_ref() - .map(|info| info.score as f32 / info.played as f32); - if let Some(score) = score { - info!("{action:?} -> {score}"); - } else { - info!("{action:?} -> None"); - } - } let actions = my_mcts_manager.base.next.map(|action| { action.map_or(0.0, |action| action.score as f32 / action.played as f32) @@ -171,7 +160,7 @@ async fn get_move(request: Json) -> response::Json { (actions, my_mcts_manager.base.played) }) }); - let (actions, played) = join_all(action_futures).await.into_iter().fold( + let (scores, played) = join_all(action_futures).await.into_iter().fold( ([0.0; 4], 0), |(mut total, mut games), actions| { if let Ok((actions, new_games)) = actions { @@ -183,7 +172,11 @@ async fn get_move(request: Json) -> response::Json { (total, games) }, ); - let action = actions + for action in actions { + let score = scores[usize::from(action)]; + info!("{action:?} -> {score}"); + } + let action = scores .into_iter() .enumerate() .max_by_key(|(_, score)| FloatOrd(*score))