Use Random.default_rng (#43)

TuringLang · Feb 20, 2023 · c2c4d8b · c2c4d8b · devmotion · Feb 20, 2023
1 parent 64ef0ee
commit c2c4d8b
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 8 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "AdvancedVI"
 uuid = "b5ca4192-6429-45e5-a2d9-87aec30a685c"
-version = "0.2.0"
+version = "0.2.1"
 
 [deps]
 Bijectors = "76274a88-744f-5084-9051-94815aaf08c4"

diff --git a/src/AdvancedVI.jl b/src/AdvancedVI.jl
@@ -1,6 +1,6 @@
 module AdvancedVI
 
-using Random: AbstractRNG
+using Random: Random
 
 using Distributions, DistributionsAD, Bijectors
 using DocStringExtensions

diff --git a/src/advi.jl b/src/advi.jl
@@ -2,7 +2,6 @@ using StatsFuns
 using DistributionsAD
 using Bijectors
 using Bijectors: TransformedDistribution
-using Random: AbstractRNG, GLOBAL_RNG
 
 
 """
@@ -53,7 +52,7 @@ end
 
 # WITHOUT updating parameters inside ELBO
 function (elbo::ELBO)(
-    rng::AbstractRNG,
+    rng::Random.AbstractRNG,
     alg::ADVI,
     q::VariationalPosterior,
     logπ::Function,

diff --git a/src/objectives.jl b/src/objectives.jl
@@ -1,9 +1,7 @@
-using Random: GLOBAL_RNG
-
 struct ELBO <: VariationalObjective end
 
 function (elbo::ELBO)(alg, q, logπ, num_samples; kwargs...)
-    return elbo(GLOBAL_RNG, alg, q, logπ, num_samples; kwargs...)
+    return elbo(Random.default_rng(), alg, q, logπ, num_samples; kwargs...)
 end
 
 const elbo = ELBO()
diff --git a/src/utils.jl b/src/utils.jl
@@ -1,6 +1,5 @@
 using Distributions
 
-using Random: Random
 using Bijectors: Bijectors