restore rng_state (#353)

NREL-Sienna · Aug 29, 2023 · 7e64147 · 7e64147
1 parent a51524d
commit 7e64147
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/base/jacobian.jl b/src/base/jacobian.jl
@@ -71,7 +71,9 @@ function JacobianFunctionWrapper(
     if sparse_retrieve_loop > 0
         for _ in 1:sparse_retrieve_loop
             temp = zeros(n, n)
+            rng_state = copy(Random.default_rng())
             Jf(temp, x0 + Random.randn(n))
+            copy!(Random.default_rng(), rng_state)
             jac .+= abs.(temp)
         end
         Jv = SparseArrays.sparse(jac)
@@ -104,7 +106,9 @@ function JacobianFunctionWrapper(
     if sparse_retrieve_loop > 0
         for _ in 1:sparse_retrieve_loop
             temp = zeros(n, n)
+            rng_state = copy(Random.default_rng())
             Jf(temp, x0 + Random.randn(n))
+            copy!(Random.default_rng(), rng_state)
             jac .+= abs.(temp)
         end
         Jv = SparseArrays.sparse(jac .+ mass_matrix)