DArray: Always return Array from collect

Author: jpsamaroo

src/array/darray.jl

@@ -194,7 +194,7 @@ function Base.collect(d::DArray{T,N}; tree=false, copyto=false) where {T,N}
     if tree
         collect(fetch(treereduce_nd(map(x -> ((args...,) -> Dagger.@spawn x(args...)) , dimcatfuncs), a.chunks)))
     else
-        treereduce_nd(dimcatfuncs, asyncmap(fetch, a.chunks))
+        collect(treereduce_nd(dimcatfuncs, asyncmap(fetch, a.chunks)))
     end
 end
 Array{T,N}(A::DArray{S,N}) where {T,N,S} = convert(Array{T,N}, collect(A))

src/array/map-reduce.jl

@@ -86,7 +86,7 @@ _mapreduce_maybesync(f, op_inner, op_outer, x, ::Colon, init) =
     _mapreduce_maybesync(f, op_inner, op_outer, x, nothing, init)
 function _mapreduce_maybesync(f, op_inner, op_outer, x::DArray{T,N}, dims::Nothing, init) where {T,N}
     Dx = _to_darray(MapReduce(f, op_inner, op_outer, x, dims, init))
-    return collect(Dx)
+    return only(collect(Dx))
 end
 
 function Base.size(r::MapReduce)