Preserve metadata when scalarizing getindex

JuliaSymbolics · Nov 12, 2023 · c056057 · c056057
1 parent fb92fd1
commit c056057
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/arrays.jl b/src/arrays.jl
@@ -727,7 +727,12 @@ function scalarize(arr)
         end
     elseif istree(arr) && operation(arr) == getindex
         args = arguments(arr)
-        scalarize(args[1], (args[2:end]...,))
+        scalarized = scalarize(args[1], (args[2:end]...,))
+        if isnothing(metadata(arr))
+            scalarized
+        else
+            metadata(scalarized, metadata(arr))
+        end
     elseif arr isa Num
         wrap(scalarize(unwrap(arr)))
     elseif istree(arr) && symtype(arr) <: Number