Add tree search node count as CLI option.

Author: kylc

python/miniacd/cli.py

@@ -60,6 +60,12 @@ def random_rgb() -> tuple[int, int, int]:
     type=click.IntRange(min=0),
     help="Monte Carlo tree search search iterations",
 )
+@click.option(
+    "--mcts-nodes",
+    default=20,
+    type=click.IntRange(min=0),
+    help="The discretization size in the Monte Carlo tree search",
+)
 @click.option(
     "--seed",
     default=42,
@@ -75,6 +81,7 @@ def main(
     threshold: float,
     mcts_depth: int,
     mcts_iterations: int,
+    mcts_nodes: int,
     seed: int,
 ):
     """
@@ -89,6 +96,7 @@ def main(
         threshold=threshold,
         max_depth=mcts_depth,
         iterations=mcts_iterations,
+        num_nodes=mcts_nodes,
         random_seed=seed,
         print=True,
     )

src/py.rs

@@ -41,13 +41,15 @@ mod pyminiacd {
                       threshold=0.1,
                       iterations=150,
                       max_depth=3,
+                      num_nodes=20,
                       random_seed=42,
                       print=true))]
     fn run(
         mesh: &PyMesh,
         threshold: f64,
         iterations: usize,
         max_depth: usize,
+        num_nodes: usize,
         random_seed: u64,
         print: bool,
     ) -> Vec<PyMesh> {
@@ -56,7 +58,7 @@ mod pyminiacd {
             iterations,
             max_depth,
             exploration_param: f64::sqrt(2.0),
-            num_nodes: 20,
+            num_nodes,
             random_seed,
             print,
         };