diff src/LazyTensors/lazy_array.jl @ 1142:6fd97b5ed3e5 bugfix/lazy_tensors/lazyfunctionarray

Fix failing test
author Jonatan Werpers <jonatan@werpers.com>
date Wed, 19 Oct 2022 23:11:59 +0200
parents c78c9a95024a
children c5cf32fab163
line wrap: on
line diff
--- a/src/LazyTensors/lazy_array.jl	Wed Oct 19 23:10:58 2022 +0200
+++ b/src/LazyTensors/lazy_array.jl	Wed Oct 19 23:11:59 2022 +0200
@@ -28,7 +28,7 @@
 export LazyFunctionArray
 
 function LazyFunctionArray(f::F, size::NTuple{D,Int}) where {F<:Function,D}
-    T = typeof(f(ones(D)...))
+    T = typeof(f(ones(Int, D)...))
     return LazyFunctionArray{F,T,D}(f,size)
 end