add test group options

TuringLang · Jun 13, 2024 · d44e36d · d44e36d · github-actions · Jun 13, 2024
1 parent 748b8a9
commit d44e36d
Showing 1 changed file with 15 additions and 9 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -18,10 +18,12 @@ using DistributionsAD
 using LogDensityProblems
 using Optimisers
 using ADTypes
-using ForwardDiff, ReverseDiff, Zygote, Enzyme
+using ForwardDiff, ReverseDiff, Zygote
 
 using AdvancedVI
 
+const GROUP = get(ENV, "GROUP", "All")
+
 # Models for Inference Tests
 struct TestModel{M,L,S}
     model::M
@@ -34,11 +36,15 @@ include("models/normal.jl")
 include("models/normallognormal.jl")
 
 # Tests
-include("interface/ad.jl")
-include("interface/optimize.jl")
-include("interface/repgradelbo.jl")
-include("interface/location_scale.jl")
-
-include("inference/repgradelbo_distributionsad.jl")
-include("inference/repgradelbo_locationscale.jl")
-include("inference/repgradelbo_locationscale_bijectors.jl")
+if GROUP == "All" || GROUP == "Interface"
+    include("interface/ad.jl")
+    include("interface/optimize.jl")
+    include("interface/repgradelbo.jl")
+    include("interface/location_scale.jl")
+end
+
+if GROUP == "All" || GROUP == "Inference"
+    include("inference/repgradelbo_distributionsad.jl")
+    include("inference/repgradelbo_locationscale.jl")
+    include("inference/repgradelbo_locationscale_bijectors.jl")
+end