update

2024-10-14 12:47:59 +07:00
parent 351bccc059
commit 3338085567
1 changed files with 2 additions and 4 deletions
--- a/src/interface.jl
+++ b/src/interface.jl
@@ -84,10 +84,8 @@ function runMCTS(
      backpropagate(leafNode, simTrajectoryReward)
    end

-    # stop if the tree early stop condition is met
-    if typeof(earlystop) <: Function
-      result = earlystop(node.state)
-
+    # stop if the early stop condition is met
+    if typeof(earlystop) <: Function && earlystop(node.state)
      break
    end
  end