Simplify test_broken syntax

TuringLang · Jan 20, 2025 · 373c894 · 373c894
1 parent 107fac5
commit 373c894
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/test/mcmc/hmc.jl b/test/mcmc/hmc.jl
@@ -86,11 +86,7 @@ using Turing
         r_mean = dropdims(mean(r; dims=1); dims=1)
 
         # TODO(mhauru) The below remains broken for Enzyme. Need to investigate why.
-        if !(adbackend isa AutoEnzyme)
-            @test isapprox(r_mean, mean(dist); atol=0.2)
-        else
-            @test_broken isapprox(r_mean, mean(dist); atol=0.2)
-        end
+        @test isapprox(r_mean, mean(dist); atol=0.2) broken=(adbackend isa AutoEnzyme)
     end
 
     @testset "multivariate support" begin