Handle array symbolics in CSE topological_sort

Co-authored-by: Aayush Sabharwal <aayush.sabharwal@gmail.com>

Author: bowenszhu

src/code.jl

@@ -727,6 +727,12 @@ function topological_sort(graph)
             push!(sorted_nodes, sym ← new_node)
             visited[node] = sym
             return sym
+        elseif _is_array_of_symbolics(node)
+            new_node = map(dfs, node)
+            sym = newsym(typeof(new_node))
+            push!(sorted_nodes, sym ← new_node)
+            visited[node] = sym
+            return sym
         else
             visited[node] = node
             return node