Also fix when dims + T are passed

Author: SouthEndMusic

src/named_array_partition.jl

@@ -49,8 +49,12 @@ end
 
 # return ArrayPartition when possible, otherwise next best thing of the correct size
 function Base.similar(A::NamedArrayPartition, ::Type{T}, dims::NTuple{N, Int}) where {T, N}
-    NamedArrayPartition(
-        similar(getfield(A, :array_partition), T, dims), getfield(A, :names_to_indices))
+    similar_array_partition = similar(getfield(A, :array_partition), T, dims)
+    if similar_array_partition isa ArrayPartition
+        NamedArrayPartition(similar_array_partition, getfield(A, :names_to_indices))
+    else
+        similar_array_partition
+    end
 end
 
 # similar array partition with different types

test/named_array_partition_tests.jl

@@ -7,6 +7,7 @@ using RecursiveArrayTools, Test
     @test typeof(similar(x)) <: NamedArrayPartition
     @test typeof(similar(x, Int)) <: NamedArrayPartition
     @test typeof(similar(x, (5, 5))) <: Matrix
+    @test typeof(similar(x, Int, (5, 5))) <: Matrix
     @test x.a ≈ ones(10)
     @test typeof(x .+ x[1:end]) <: Vector # test broadcast precedence 
     @test all(x .== x[1:end])