Update vectorized_product_iterator.jl

src/Containers/vectorized_product_iterator.jl

@@ -45,16 +45,13 @@ struct VectorizedProductIterator{T}
     prod::Iterators.ProductIterator{T}
 end
 
-function _collect(x)
-    # Collect iterators with unknown size so they can be used as axes.
-    if Base.IteratorSize(x) == Base.SizeUnknown()
-        return collect(x)
-    elseif Base.IteratorSize(x) == Base.IsInfinite()
-        error("Unable to form a container. Axis $(x) has infinite size!")
-    else
-        return x
-    end
+# Collect iterators with unknown size so they can be used as axes.
+_collect(::Base.SizeUnknown, x) = collect(x)
+_collect(::Any, x) = x
+function _collect(::Base.IsInfinite, x)
+    return error("Unable to form a container. Axis $(x) has infinite size!")
 end
+_collect(x) = _collect(Base.IteratorSize(x), x)
 
 function vectorized_product(iterators...)
     return VectorizedProductIterator(Iterators.product(_collect.(iterators)...))