maybe now

TuringLang · Apr 25, 2024 · 7f7ba11 · 7f7ba11
1 parent d561f59
commit 7f7ba11
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/ext/AdvancedVIEnzymeExt.jl b/ext/AdvancedVIEnzymeExt.jl
@@ -4,7 +4,7 @@ if isdefined(Base, :get_extension)
     using AdvancedVI: AdvancedVI, ADTypes, DiffResults, Distributions
     using Enzyme: Enzyme
 else
-    using ..AdvancedVI: ADTypes, AdvancedVI
+    using ..AdvancedVI: AdvancedVI, ADTypes, DiffResults, Distributions
     using ..Enzyme: Enzyme
 end
 

diff --git a/ext/AdvancedVIReverseDiffExt.jl b/ext/AdvancedVIReverseDiffExt.jl
@@ -4,7 +4,7 @@ if isdefined(Base, :get_extension)
     using AdvancedVI: AdvancedVI, ADTypes, DiffResults, Distributions
     using ReverseDiff: ReverseDiff
 else
-    using ..AdvancedVI: ADTypes, AdvancedVI
+    using ..AdvancedVI: AdvancedVI, ADTypes, DiffResults, Distributions
     using ..ReverseDiff: ReverseDiff
 end
 

diff --git a/ext/AdvancedVIZygoteExt.jl b/ext/AdvancedVIZygoteExt.jl
@@ -4,7 +4,7 @@ if isdefined(Base, :get_extension)
     using AdvancedVI: AdvancedVI, ADTypes, DiffResults, Distributions
     using Zygote: Zygote
 else
-    using ..AdvancedVI: ADTypes, AdvancedVI
+    using ..AdvancedVI: AdvancedVI, ADTypes, DiffResults, Distributions
     using ..Zygote: Zygote
 end
 

diff --git a/src/AdvancedVI.jl b/src/AdvancedVI.jl
@@ -193,7 +193,7 @@ end
         @require ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267" include(
             "../ext/AdvancedVIReverseDiffExt.jl"
         )
-        @require ZygoteRules = "700de1a5-db45-46bc-99cf-38207098b444" include(
+        @require Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f" include(
             "../ext/AdvancedVIZygoteExt.jl"
         )
         @require Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9" include(