fix conditionally load Enzyme in tests

TuringLang · Nov 30, 2024 · 870778b · 870778b
1 parent ecb4813
commit 870778b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -21,12 +21,16 @@ using DistributionsAD
 @functor TuringDiagMvNormal
 
 using ADTypes
-using ForwardDiff, ReverseDiff, Zygote, Mooncake, Enzyme
+using ForwardDiff, ReverseDiff, Zygote, Mooncake
 
 using AdvancedVI
 
 const TEST_GROUP = get(ENV, "TEST_GROUP", "All")
 
+if TEST_GROUP == "Enzyme"
+    using Enzyme
+end
+
 # Models for Inference Tests
 struct TestModel{M,L,S,SC}
     model::M