update

src/mcts.jl

@@ -196,11 +196,6 @@ end
 # Return
   - None
 
-# Example
-```jldoctest
-julia> 
-```
-
 # Signature
 """
 function expand(node::MCTSNode,transition::Function, transitionargs::NamedTuple; 
@@ -249,13 +244,14 @@ end
     Total number to sample from the current node (i.e. expand new node horizontally)
     
 # Return
-  - `(simTrajectoryReward, terminalstate)::Union{Tuple{Number, Dict{Symbol, <:Any}}, Tuple{Number, Nothing}}`
+  - `::NamedTuple{(:simTrajectoryReward, :terminalstate), Tuple{Number, Union{Dict{Symbol, Any}, Nothing}}}`
 
 # Signature
 """
 function simulate(node::MCTSNode, transition::Function, transitionargs::NamedTuple; 
   maxdepth::Integer=3, totalsample::Integer=3
-  )::Union{Tuple{Number, Dict{Symbol, <:Any}}, Tuple{Number, Nothing}}
+)::NamedTuple{(:simTrajectoryReward, :terminalstate), Tuple{Number, Union{Dict{Symbol, Any}, Nothing}}}
+
   simTrajectoryReward = 0.0
   terminalstate = nothing
   
@@ -271,7 +267,7 @@ function simulate(node::MCTSNode, transition::Function, transitionargs::NamedTup
     end
   end
 
-  return (simTrajectoryReward, terminalstate)
+  return (simTrajectoryReward=simTrajectoryReward, terminalstate=terminalstate)
 end