restore MLUtils imports

gher-uliege · Oct 2, 2024 · cd083e2 · cd083e2
1 parent f61323e
commit cd083e2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/diffusion_model.jl b/src/diffusion_model.jl
@@ -246,6 +246,7 @@ struct DatasetLoader{T,N,Trng,Tdevice,Ta,Taux}
     training::Bool
 end
 
+import MLUtils: numobs, getobs, getobs!
 numobs(d::DatasetLoader) = size(d.train_input)[end]
 
 function getobs_orig(d::DatasetLoader,index::Union{AbstractVector,Integer})