Skip to content
Snippets Groups Projects
Commit b0000e16 authored by Fredrik Bagge Carlson's avatar Fredrik Bagge Carlson
Browse files

remove enumerate

parent e0ffc590
Branches
No related tags found
No related merge requests found
......@@ -111,8 +111,10 @@ function collect_episode(ep, max_size=1000)
rewards = Float64[]
actions = Int[]
raw_frames = Array{Float64,2}(100800,max_size+1)
local s1, i
for (i,(s,a,r,s1)) in take(enumerate(ep), max_size)
local s1
i = 0
for (s,a,r,s1) in take(ep, max_size)
i += 1
raw_frames[:,i] = s
if (i-1) % action_repeat == 0
push!(rewards, r)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment