fix test rng

Author: prbzrg

test/ci_tests/regression_tests.jl

@@ -1,8 +1,9 @@
 Test.@testset "Regression Tests" begin
+    rng = StableRNGs.StableRNG(1)
     ndata = 2^10
     ndimension = 1
     data_dist = Distributions.Beta{Float32}(2.0f0, 4.0f0)
-    r = rand(data_dist, ndimension, ndata)
+    r = rand(rng, data_dist, ndimension, ndata)
     r = convert.(Float32, r)
 
     nvars = size(r, 1)
@@ -22,6 +23,7 @@ Test.@testset "Regression Tests" begin
         λ₁ = 1.0f-2,
         λ₂ = 1.0f-2,
         λ₃ = 1.0f-2,
+        rng,
         sol_kwargs = (;
             save_everystep = false,
             alg = OrdinaryDiffEqDefault.DefaultODEAlgorithm(),

test/ci_tests/speed_tests.jl

@@ -21,10 +21,11 @@ Test.@testset "Speed Tests" begin
     Test.@testset "$compute_mode" for compute_mode in compute_modes
         @show compute_mode
 
+        rng = StableRNGs.StableRNG(1)
         ndata = 2^10
         ndimension = 1
         data_dist = Distributions.Beta{Float32}(2.0f0, 4.0f0)
-        r = rand(data_dist, ndimension, ndata)
+        r = rand(rng, data_dist, ndimension, ndata)
         r = convert.(Float32, r)
 
         nvars = size(r, 1)
@@ -44,6 +45,7 @@ Test.@testset "Speed Tests" begin
             λ₁ = 1.0f-2,
             λ₂ = 1.0f-2,
             λ₃ = 1.0f-2,
+            rng,
             sol_kwargs = (;
                 save_everystep = false,
                 alg = OrdinaryDiffEqDefault.DefaultODEAlgorithm(),
@@ -52,7 +54,6 @@ Test.@testset "Speed Tests" begin
         )
 
         df = DataFrames.DataFrame(transpose(r), :auto)
-
         model = ContinuousNormalizingFlows.ICNFModel(
             icnf;
             batchsize = 0,

test/runtests.jl

@@ -31,8 +31,7 @@ if GROUP == "All"
     Logging.global_logger(debuglogger)
 end
 
-Test.@testset verbose = true showtiming = true failfast = false rng =
-    StableRNGs.StableRNG(1) "Overall" begin
+Test.@testset verbose = true showtiming = true failfast = false "Overall" begin
     if GROUP == "All" || GROUP in ["SmokeXOut", "SmokeXIn", "SmokeXYOut", "SmokeXYIn"]
         include(joinpath("ci_tests", "smoke_tests.jl"))
     end