_buffer_frame for Cartesian index

JuliaReinforcementLearning · Aug 7, 2023 · c78e3b5 · c78e3b5
1 parent 45f79d4
commit c78e3b5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/CircularArrayBuffers.jl b/src/CircularArrayBuffers.jl
@@ -96,6 +96,7 @@ end
     end
 end
 
+_buffer_frame(cb::CircularArrayBuffer, I::CartesianIndex) = map(i -> _buffer_frame(cb, i), Tuple(I)...)
 _buffer_frame(cb::CircularArrayBuffer, I::AbstractArray{<:Integer}) = map(i -> _buffer_frame(cb, i), I)
 _buffer_frame(cb::CircularArrayBuffer, ::Colon) = _buffer_frame(cb, axes(cb)[end])