fix: pass in RNG to shuffle

LuxDL · Jan 7, 2025 · 4125770 · 4125770
1 parent d2dc63f
commit 4125770
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/WeightInitializers/Project.toml b/lib/WeightInitializers/Project.toml
@@ -1,7 +1,7 @@
 name = "WeightInitializers"
 uuid = "d49dbf32-c5c2-4618-8acc-27bb2598ef2d"
 authors = ["Avik Pal <[email protected]> and contributors"]
-version = "1.0.4"
+version = "1.0.5"
 
 [deps]
 ArgCheck = "dce04be8-c92d-5529-be00-80e4d2c0e197"

diff --git a/lib/WeightInitializers/src/initializers.jl b/lib/WeightInitializers/src/initializers.jl
@@ -225,7 +225,7 @@ function sparse_init(rng::AbstractRNG, ::Type{T}, dims::Integer...;
     sparse_array .*= T(std)
     fill!(view(sparse_array, 1:num_zeros, :), zero(T))
 
-    return @allowscalar mapslices(shuffle, sparse_array; dims=1)
+    return @allowscalar mapslices(Base.Fix1(shuffle, rng), sparse_array; dims=1)
 end
 
 """