diff --git a/test/runtests.jl b/test/runtests.jl index 8e0cbc11..6a0bf7af 100644 --- 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