fix: fix inference of getindex(::ArrayPartition, ::Int)

SciML · Feb 13, 2024 · 765c29c · 765c29c
1 parent 3543a1b
commit 765c29c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/array_partition.jl b/src/array_partition.jl
@@ -219,6 +219,7 @@ Base.@propagate_inbounds function Base.getindex(A::ArrayPartition, i::Int)
             return A.x[j][length(A.x[j]) + i]
         end
     end
+    throw(BoundsError(A, i))
 end
 
 """

diff --git a/test/partitions_test.jl b/test/partitions_test.jl
@@ -1,6 +1,7 @@
 using RecursiveArrayTools, Test, Statistics, ArrayInterface
 A = (rand(5), rand(5))
 p = ArrayPartition(A)
+@inferred p[1]
 @test (p.x[1][1], p.x[2][1]) == (p[1], p[6])
 
 p = ArrayPartition(A, Val{true})
-Original file line number
+Diff line change
@@ Expand Up @@
                 return A.x[j][length(A.x[j]) + i]
             end
         end
+        throw(BoundsError(A, i))
     end
     """
@@ Expand Down @@