catch empty rollout

elevate-lang · Oct 22, 2024 · 034b4ad · 034b4ad
1 parent 23baf72
commit 034b4ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/scala/elevate/heuristic_search/heuristics/MCTS.scala b/src/main/scala/elevate/heuristic_search/heuristics/MCTS.scala
@@ -103,7 +103,7 @@ class MCTS[P] extends Heuristic[P] {
           rollout = choose_valid_solution_randomly(panel = panel, actions = actions, rollout._2)
 
           // check if rollout is empty
-          if (rollout == null) {
+          if (rollout._1 == null) {
             isTerminal = true
           } else {
             rollout._1.solutionSteps.foreach(step => println(s"""[${step.strategy}, ${step.location}]"""))