fix

Author: Jeremiah Lewis

src/common/ElasticArraySARTSTraces.jl

@@ -21,10 +21,10 @@ function ElasticArraySARTSTraces(;
     reward_eltype, reward_size = reward
     terminal_eltype, terminal_size = terminal
 
-    MultiplexTraces{SS′}(ElasticArray{state_eltype}(state_size..., Inf)) +
+    MultiplexTraces{SS′}(ElasticArray{state_eltype}(undef, state_size..., 0)) +
     Traces(
-        action = ElasticArray{action_eltype}(action_size..., Inf),
-        reward=ElasticArray{reward_eltype}(reward_size..., Inf),
-        terminal=ElasticArray{terminal_eltype}(terminal_size..., Inf),
+        action = ElasticArray{action_eltype}(undef, action_size..., 0),
+        reward=ElasticArray{reward_eltype}(undef, reward_size..., 0),
+        terminal=ElasticArray{terminal_eltype}(undef, terminal_size..., 0),
     )
 end

src/common/ElasticArraySLARTTraces.jl

@@ -25,11 +25,11 @@ function ElasticArraySLARTTraces(;
     reward_eltype, reward_size = reward
     terminal_eltype, terminal_size = terminal
 
-    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)) +
+    MultiplexTraces{SS′}(ElasticArray{state_eltype}(undef, state_size..., 0)) +
+    MultiplexTraces{LL′}(ElasticArray{legal_actions_mask_eltype}(undef, legal_actions_mask_size..., 0)) +
+    MultiplexTraces{AA′}(ElasticArray{action_eltype}(undef, action_size..., 0)) +
     Traces(
-        reward=ElasticArray{reward_eltype}(reward_size..., capacity),
-        terminal=ElasticArray{terminal_eltype}(terminal_size..., capacity),
+        reward=ElasticArray{reward_eltype}(undef, reward_size..., 0),
+        terminal=ElasticArray{terminal_eltype}(undef, terminal_size..., 0),        
     )
 end

test/episodes.jl

@@ -1,6 +1,7 @@
 using ReinforcementLearningTrajectories
 using CircularArrayBuffers
 using Test
+
 @testset "EpisodesBuffer" begin
     @testset "with circular traces" begin 
         eb = EpisodesBuffer(
@@ -282,7 +283,7 @@ using Test
     end
     @testset "with PartialNamedTuple" begin 
         eb = EpisodesBuffer(
-            CircularArraySARTSATraces(;
+            ElasticArraySARTSATraces(;
             capacity=10)
         )
         #push a first episode l=5