fix

Author: Jeremiah Lewis

Project.toml

@@ -1,6 +1,6 @@
 name = "ReinforcementLearningTrajectories"
 uuid = "6486599b-a3cd-4e92-a99a-2cea90cc8c3c"
-version = "0.3.7"
+version = "0.4"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

src/common/ElasticArraySARTSTraces.jl

@@ -3,10 +3,10 @@ export ElasticArraySARTSTraces
 const ElasticArraySARTSTraces = Traces{
     SS′ART,
     <:Tuple{
-        <:MultiplexTraces{SS′,<:Trace{<:ElasticArrayBuffer}},
-        <:Trace{<:ElasticArrayBuffer},
-        <:Trace{<:ElasticArrayBuffer},
-        <:Trace{<:ElasticArrayBuffer},
+        <:MultiplexTraces{SS′,<:Trace{<:ElasticArray}},
+        <:Trace{<:ElasticArray},
+        <:Trace{<:ElasticArray},
+        <:Trace{<:ElasticArray},
     }
 }
 
@@ -21,10 +21,10 @@ function ElasticArraySARTSTraces(;
     reward_eltype, reward_size = reward
     terminal_eltype, terminal_size = terminal
 
-    MultiplexTraces{SS′}(ElasticArrayBuffer{state_eltype}(state_size..., capacity+1)) +
+    MultiplexTraces{SS′}(ElasticArray{state_eltype}(state_size..., Inf)) +
     Traces(
-        action = ElasticArrayBuffer{action_eltype}(action_size..., capacity),
-        reward=ElasticArrayBuffer{reward_eltype}(reward_size..., capacity),
-        terminal=ElasticArrayBuffer{terminal_eltype}(terminal_size..., capacity),
+        action = ElasticArray{action_eltype}(action_size..., Inf),
+        reward=ElasticArray{reward_eltype}(reward_size..., Inf),
+        terminal=ElasticArray{terminal_eltype}(terminal_size..., Inf),
     )
 end

src/common/ElasticArraySLARTTraces.jl

@@ -3,11 +3,11 @@ export ElasticArraySLARTTraces
 const ElasticArraySLARTTraces = Traces{
     SS′LL′AA′RT,
     <:Tuple{
-        <:MultiplexTraces{SS′,<:Trace{<:ElasticArrayBuffer}},
-        <:MultiplexTraces{LL′,<:Trace{<:ElasticArrayBuffer}},
-        <:MultiplexTraces{AA′,<:Trace{<:ElasticArrayBuffer}},
-        <:Trace{<:ElasticArrayBuffer},
-        <:Trace{<:ElasticArrayBuffer},
+        <:MultiplexTraces{SS′,<:Trace{<:ElasticArray}},
+        <:MultiplexTraces{LL′,<:Trace{<:ElasticArray}},
+        <:MultiplexTraces{AA′,<:Trace{<:ElasticArray}},
+        <:Trace{<:ElasticArray},
+        <:Trace{<:ElasticArray},
     }
 }
 
@@ -25,11 +25,11 @@ function ElasticArraySLARTTraces(;
     reward_eltype, reward_size = reward
     terminal_eltype, terminal_size = terminal
 
-    MultiplexTraces{SS′}(ElasticArrayBuffer{state_eltype}(state_size..., capacity + 1)) +
-    MultiplexTraces{LL′}(ElasticArrayBuffer{legal_actions_mask_eltype}(legal_actions_mask_size..., capacity + 1)) +
-    MultiplexTraces{AA′}(ElasticArrayBuffer{action_eltype}(action_size..., capacity + 1)) +
+    MultiplexTraces{SS′}(ElasticArray{state_eltype}(state_size..., capacity + 1)) +
+    MultiplexTraces{LL′}(ElasticArray{legal_actions_mask_eltype}(legal_actions_mask_size..., capacity + 1)) +
+    MultiplexTraces{AA′}(ElasticArray{action_eltype}(action_size..., capacity + 1)) +
     Traces(
-        reward=ElasticArrayBuffer{reward_eltype}(reward_size..., capacity),
-        terminal=ElasticArrayBuffer{terminal_eltype}(terminal_size..., capacity),
+        reward=ElasticArray{reward_eltype}(reward_size..., capacity),
+        terminal=ElasticArray{terminal_eltype}(terminal_size..., capacity),
     )
 end

src/common/ElasticPrioritizedTraces.jl

@@ -1,7 +1,9 @@
 export ElasticPrioritizedTraces
 
+using ElasticArrays: ElasticVector
+
 struct ElasticPrioritizedTraces{T,names,Ts} <: AbstractTraces{names,Ts}
-    keys::ElasticVectorBuffer{Int,Vector{Int}}
+    keys::ElasticVector{Int,Vector{Int}}
     priorities::SumTree{Float32}
     traces::T
     default_priority::Float32
@@ -12,7 +14,7 @@ function ElasticPrioritizedTraces(traces::AbstractTraces{names,Ts}; default_prio
     new_Ts = Tuple{Int,Float32,Ts.parameters...}
     c = capacity(traces)
     ElasticPrioritizedTraces{typeof(traces),new_names,new_Ts}(
-        ElasticVectorBuffer{Int}(c),
+        ElasticVector{Int}(c),
         SumTree(c),
         traces,
         default_priority

src/episodes.jl

@@ -1,6 +1,6 @@
 export EpisodesBuffer, PartialNamedTuple
 import DataStructures.CircularBuffer
-using ElasticArrays: ElasticArray, ElasticVector, resize_lastdim!
+using ElasticArrays: ElasticArray, ElasticVector
 
 """
     EpisodesBuffer(traces::AbstractTraces)