improve progressmeter interface in inference tests

TuringLang · Jun 13, 2024 · 47aba44 · 47aba44
1 parent d44e36d
commit 47aba44
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 12 deletions.
diff --git a/test/inference/repgradelbo_distributionsad.jl b/test/inference/repgradelbo_distributionsad.jl
@@ -1,8 +1,4 @@
 
-const PROGRESS = length(ARGS) > 0 && ARGS[1] == "--progress" ? true : false
-
-using Test
-
 @testset "inference RepGradELBO DistributionsAD" begin
     @testset "$(modelname) $(objname) $(realtype) $(adbackname)"  for
         realtype ∈ [Float64, Float32],

diff --git a/test/inference/repgradelbo_locationscale.jl b/test/inference/repgradelbo_locationscale.jl
@@ -1,8 +1,4 @@
 
-const PROGRESS = length(ARGS) > 0 && ARGS[1] == "--progress" ? true : false
-
-using Test
-
 @testset "inference RepGradELBO VILocationScale" begin
     @testset "$(modelname) $(objname) $(realtype) $(adbackname)"  for
         realtype in [Float64, Float32],

diff --git a/test/inference/repgradelbo_locationscale_bijectors.jl b/test/inference/repgradelbo_locationscale_bijectors.jl
@@ -1,8 +1,4 @@
 
-const PROGRESS = length(ARGS) > 0 && ARGS[1] == "--progress" ? true : false
-
-using Test
-
 @testset "inference RepGradELBO VILocationScale Bijectors" begin
     @testset "$(modelname) $(objname) $(realtype) $(adbackname)"  for
         realtype in [Float64, Float32],

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -43,6 +43,9 @@ if GROUP == "All" || GROUP == "Interface"
     include("interface/location_scale.jl")
 end
 
+
+const PROGRESS = haskey(ENV, "PROGRESS")
+
 if GROUP == "All" || GROUP == "Inference"
     include("inference/repgradelbo_distributionsad.jl")
     include("inference/repgradelbo_locationscale.jl")