Merge branch 'master' of github.com:findmyway/CircularArrayBuffers.jl

Author: findmyway

Project.toml

@@ -1,7 +1,7 @@
 name = "CircularArrayBuffers"
 uuid = "9de3a189-e0c0-4e15-ba3b-b14b9fb0aec1"
 authors = ["Jun Tian <tianjun.cpp@gmail.com> and contributors"]
-version = "0.1.3"
+version = "0.1.4"
 
 [compat]
 julia = "1"

src/CircularArrayBuffers.jl

@@ -72,7 +72,11 @@ function Base.push!(cb::CircularArrayBuffer{T,N}, data) where {T,N}
         cb.nframes += 1
     end
     if N == 1
-        cb[cb.nframes] = data
+        if ndims(data) == 0
+            cb[cb.nframes] = data[]
+        else
+            cb[cb.nframes] = data
+        end
     else
         cb[ntuple(_ -> (:), N - 1)..., cb.nframes] .= data
     end

test/runtests.jl

@@ -4,6 +4,15 @@ using Test
 @testset "CircularArrayBuffers.jl" begin
     A = ones(2, 2)
     C = ones(Float32, 2, 2)
+
+    # https://github.com/JuliaReinforcementLearning/ReinforcementLearning.jl/issues/551
+    @testset "1D with 0d data" begin
+        b = CircularArrayBuffer{Int}(3)
+        push!(b, zeros(Int, ()))
+        @test length(b) == 1
+        @test b[1] == 0
+    end
+
     @testset "1D Int" begin
         b = CircularArrayBuffer{Int}(3)