Fix EpisodesBuffer being trace capacity + 2 instead of + 1

Author: ludvigk

src/episodes.jl

@@ -54,7 +54,8 @@ function is_capacity_plus_one(traces::AbstractTraces)
 end
 
 function EpisodesBuffer(traces::AbstractTraces)
-    cap = is_capacity_plus_one(traces) ? capacity(traces) + 1 : capacity(traces)
+    # cap = is_capacity_plus_one(traces) ? capacity(traces) + 1 : capacity(traces)
+    cap = capacity(traces)
     @assert isempty(traces) "EpisodesBuffer must be initialized with empty traces."
     if !isinf(cap)
         legalinds =  CircularBuffer{Bool}(cap)