Fix ADTypes extension in Julia < 1.9 (#19)

tpapp · Jul 24, 2023 · e13061f · e13061f · devmotion · Jul 24, 2023
1 parent d83793c
commit e13061f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "LogDensityProblemsAD"
 uuid = "996a588d-648d-4e1f-a8f0-a84b347e47b1"
 authors = ["Tamás K. Papp <[email protected]>"]
-version = "1.6.0"
+version = "1.6.1"
 
 [deps]
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"

diff --git a/ext/LogDensityProblemsADADTypesExt.jl b/ext/LogDensityProblemsADADTypesExt.jl
@@ -1,7 +1,12 @@
 module LogDensityProblemsADADTypesExt
 
-import LogDensityProblemsAD
-import ADTypes
+if isdefined(Base, :get_extension)
+    import LogDensityProblemsAD
+    import ADTypes
+else
+    import ..LogDensityProblemsAD
+    import ..ADTypes
+end
 
 """
     ADgradient(ad::ADTypes.AbstractADType, ℓ)