henri/trictrac
1
1
Fork 0
Code Issues 13 Pull requests Projects 1 Releases Packages Wiki Activity Actions

feat: ai strategy (wip)

Browse source
This commit is contained in:
Henri Bourcereau 2025-03-02 15:20:24 +01:00
parent 899a690869
commit ab770f3a34
14 changed files with 421 additions and 57 deletions
Download patch file Download diff file Expand all files Collapse all files

9
client_cli/src/app.rs
View file

@ -1,4 +1,4 @@
use bot::{BotStrategy, DefaultStrategy};
use bot::{BotStrategy, DefaultStrategy, StableBaselines3Strategy};
use itertools::Itertools;
use crate::game_runner::GameRunner;
@ -32,6 +32,13 @@ impl App {
"dummy" => {
Some(Box::new(DefaultStrategy::default()) as Box<dyn BotStrategy>)
}
"ai" => {
Some(Box::new(StableBaselines3Strategy::default()) as Box<dyn BotStrategy>)
}
s if s.starts_with("ai:") => {
let path = s.trim_start_matches("ai:");
Some(Box::new(StableBaselines3Strategy::new(path)) as Box<dyn BotStrategy>)
}
_ => None,
})
.collect()