Merge pull request #13 from findmyway/change_push

change the default implementation of push! for trajectory
JuliaReinforcementLearning · Jan 31, 2020 · 7677fa1 · 7677fa1
2 parents 95a81c9 + 160f035
commit 7677fa1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/interface.jl b/src/interface.jl
@@ -121,11 +121,13 @@ The length of `names` and `types` must match.
 end
 
 @interface function Base.push!(t::AbstractTrajectory; kwargs...)
-    for (k, v) in kwargs
-        push!(get_trace(t, k), v)
+    for kv in kwargs
+        push!(t, kv)
     end
 end
 
+@interface Base.push!(t::AbstractTrajectory, kv::Pair{Symbol})
+
 """
     extract_transitions(trajectory::AbstractTrajectory, learner::AbstractLearner)