fix case of unknown extents

JuliaSymbolics · Oct 8, 2023 · 678b9a9 · 678b9a9
1 parent 8217a73
commit 678b9a9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/arrays.jl b/src/arrays.jl
@@ -215,7 +215,9 @@ function make_shape(output_idx, expr, ranges=Dict())
             end
             mi = matches[i]
             @assert !isempty(mi)
-            return Base.OneTo(length(get_extents(mi)))
+            ext = get_extents(mi)
+            ext isa Unknown && return Unknown()
+            return Base.OneTo(length(ext))
         elseif i isa Integer
             return Base.OneTo(1)
         end