Parse int args

primenumber · May 15, 2024 · c40456d · c40456d
1 parent 15846c7
commit c40456d
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 6 deletions.
diff --git a/src/play.rs b/src/play.rs
@@ -242,8 +242,16 @@ fn generate_depth_n(board: Board, depth: usize, prev_passed: bool, record: &mut
 
 pub fn parallel_self_play(matches: &ArgMatches) {
     let output_path = matches.get_one::<String>("OUTPUT").unwrap();
-    let random_depth = *matches.get_one::<usize>("DEPTH").unwrap();
-    let take_count = *matches.get_one::<usize>("COUNT").unwrap();
+    let random_depth = matches
+        .get_one::<String>("DEPTH")
+        .unwrap()
+        .parse::<usize>()
+        .unwrap();
+    let take_count = matches
+        .get_one::<String>("COUNT")
+        .unwrap()
+        .parse::<usize>()
+        .unwrap();
 
     let out_f = File::create(output_path).unwrap();
     let mut writer = BufWriter::new(out_f);

diff --git a/src/train.rs b/src/train.rs
@@ -42,7 +42,11 @@ pub fn clean_record(matches: &ArgMatches) {
 pub fn gen_dataset(matches: &ArgMatches) {
     let input_path = matches.get_one::<String>("INPUT").unwrap();
     let output_path = matches.get_one::<String>("OUTPUT").unwrap();
-    let max_output = *matches.get_one::<usize>("MAX_OUT").unwrap();
+    let max_output = matches
+        .get_one::<String>("MAX_OUT")
+        .unwrap()
+        .parse::<usize>()
+        .unwrap();
 
     eprintln!("Parse input...");
     let mut boards_with_results = Vec::new();
@@ -232,9 +236,21 @@ const PATTERNS_LARGE: [u64; 10] = [
 pub fn train(matches: &ArgMatches) -> Option<()> {
     let input_path = matches.get_one::<String>("INPUT").unwrap();
     let output_path = matches.get_one::<String>("OUTPUT").unwrap();
-    let range_min = *matches.get_one::<i8>("from").unwrap();
-    let range_max = *matches.get_one::<i8>("to").unwrap();
-    let width = matches.get_one::<i8>("width").unwrap();
+    let range_min = matches
+        .get_one::<String>("from")
+        .unwrap()
+        .parse::<i8>()
+        .unwrap();
+    let range_max = matches
+        .get_one::<String>("to")
+        .unwrap()
+        .parse::<i8>()
+        .unwrap();
+    let width = matches
+        .get_one::<String>("width")
+        .unwrap()
+        .parse::<i8>()
+        .unwrap();
 
     let in_f = File::open(input_path).ok()?;
     let mut reader = BufReader::new(in_f);