Fix formatting again

compintell · Oct 4, 2024 · 74f0c88 · 74f0c88
1 parent d5366bb
commit 74f0c88
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 14 deletions.
diff --git a/ext/MooncakeJETExt.jl b/ext/MooncakeJETExt.jl
@@ -1,7 +1,8 @@
 module MooncakeJETExt
 
- using JET, Mooncake
+using JET, Mooncake
+
+Mooncake.TestUtils.test_opt(::Mooncake.TestUtils.Shim, args...) = JET.test_opt(args...)
+Mooncake.TestUtils.report_opt(::Mooncake.TestUtils.Shim, tt) = JET.report_opt(tt)
 
- Mooncake.TestUtils.test_opt(::Mooncake.TestUtils.Shim, args...) = JET.test_opt(args...)
- Mooncake.TestUtils.report_opt(::Mooncake.TestUtils.Shim, tt) = JET.report_opt(tt)
 end
diff --git a/ext/MooncakeLogDensityProblemsADExt.jl b/ext/MooncakeLogDensityProblemsADExt.jl
@@ -3,17 +3,10 @@
 
 module MooncakeLogDensityProblemsADExt
 
-if isdefined(Base, :get_extension)
- using ADTypes
- using LogDensityProblemsAD: ADGradientWrapper
- import LogDensityProblemsAD: ADgradient, logdensity_and_gradient, dimension, logdensity
- import Mooncake
-else
- using ADTypes
- using ..LogDensityProblemsAD: ADGradientWrapper
- import ..LogDensityProblemsAD: ADgradient, logdensity_and_gradient, dimension, logdensity
- import ..Mooncake
-end
+using ADTypes
+using LogDensityProblemsAD: ADGradientWrapper
+import LogDensityProblemsAD: ADgradient, logdensity_and_gradient, dimension, logdensity
+import Mooncake
 
 struct MooncakeGradientLogDensity{Trule, L} <: ADGradientWrapper
  rule::Trule