Define recursive_eltype

Author: devmotion

src/array_partition.jl

@@ -182,10 +182,12 @@ function recursivecopy!(A::ArrayPartition, B::ArrayPartition)
   end
 end
 
-recursive_one(A::ArrayPartition) = recursive_one(first(A.x))
-
 recursive_mean(A::ArrayPartition) = mean((recursive_mean(x) for x in A.x))
 
+# note: consider only first partition for recursive one and eltype
+recursive_one(A::ArrayPartition) = recursive_one(first(A.x))
+recursive_eltype(A::ArrayPartition) = recursive_eltype(first(A.x))
+
 ## iteration
 
 Base.start(A::ArrayPartition) = start(Chain(A.x))

test/partitions_test.jl

@@ -60,6 +60,11 @@ x = ArrayPartition([1, 2], [3.0, 4.0])
 @inferred first(x)
 @inferred last(x)
 
+# recursive
+@inferred recursive_mean(x)
+@inferred recursive_one(x)
+@inferred recursive_eltype(x)
+
 # broadcasting
 _scalar_op(y) = y + 1
 # Can't do `@inferred(_scalar_op.(x))` so we wrap that in a function: