Add ndims def in array function registration to help infer Arr type

JuliaSymbolics · Aug 9, 2024 · e38bc90 · e38bc90
1 parent 3fde06d
commit e38bc90
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/arrays.jl b/test/arrays.jl
@@ -405,6 +405,7 @@ end
     @register_array_symbolic bar(x::AbstractVector, p::AbstractMatrix) begin
         size = size(x)
         eltype = promote_type(eltype(x), eltype(p))
+        ndims = 1
     end
 
     @test isequal(substitute(bar(x, p), x => ones(3)), bar(ones(3), p))