Merge pull request #62 from mattuntergassmair/smallfix

missed one bug in TMazes
JuliaPOMDP · Dec 18, 2019 · 409640a · 409640a
2 parents 75eeec6 + 727f071
commit 409640a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/TMazes.jl b/src/TMazes.jl
@@ -51,7 +51,7 @@ function create_transition_distribution(::TMaze)
     rp = [0.5, 0.5]
     TMazeStateDistribution(TMazeState(), false, rs, rp)
 end
-support(d::TMazeStateDistribution) = d.reset ? (return [d.current_state]) : (return zip(d.reset_states, d.reset_probs))
+support(d::TMazeStateDistribution) = d.reset ? [d.current_state] : d.reset_states
 
 function pdf(d::TMazeStateDistribution, s::TMazeState)
     if d.reset