tweak pycall return type for 1.11

slimgroup · Oct 17, 2024 · aa1b868 · aa1b868
1 parent e238776
commit aa1b868
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/TimeModeling/Modeling/python_interface.jl b/src/TimeModeling/Modeling/python_interface.jl
@@ -35,9 +35,10 @@ function wrapcall_weights(func, args...;kw...)
     return out
 end
 
-function wrapcall_wf(func, args...;kw...)
-    rtype = _outtype(get(kw, :illum, nothing), 1, Array{Float32})
-    out = rlock_pycall(func, rtype, args...;kw...)
+function wrapcall_wf(func, modelPy, args...;kw...)
+    ndim = modelPy.dim
+    rtype = _outtype(get(kw, :illum, nothing), 1, Array{Float32, ndim})
+    out = rlock_pycall(func, rtype, modelPy, args...;kw...)
     return out
 end