From 930cd6d49e8d8099fb163ed2983c1e5941acc89a Mon Sep 17 00:00:00 2001
From: Kyurae Kim <kyrkim@seas.upenn.edu>
Date: Fri, 4 Oct 2024 12:33:11 -0700
Subject: [PATCH] fix wrong mooncake API

---
 bench/benchmarks.jl                                   | 2 +-
 test/inference/repgradelbo_distributionsad.jl         | 2 +-
 test/inference/repgradelbo_locationscale.jl           | 2 +-
 test/inference/repgradelbo_locationscale_bijectors.jl | 2 +-
 test/inference/scoregradelbo_distributionsad.jl       | 2 +-
 test/inference/scoregradelbo_locationscale.jl         | 2 +-
 test/interface/ad.jl                                  | 2 +-
 test/interface/repgradelbo.jl                         | 2 +-
 8 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/bench/benchmarks.jl b/bench/benchmarks.jl
index 6b06d8d8..22daaaff 100644
--- a/bench/benchmarks.jl
+++ b/bench/benchmarks.jl
@@ -50,7 +50,7 @@ begin
                 ("Zygote", AutoZygote()),
                 ("ForwardDiff", AutoForwardDiff()),
                 ("ReverseDiff", AutoReverseDiff()),
-                ("Mooncake", AutoMooncake(; config=Mooncake.config())),
+                ("Mooncake", AutoMooncake(; config=Mooncake.Config())),
                 #("Enzyme", AutoEnzyme()),
             ],
             (familyname, family) in [
diff --git a/test/inference/repgradelbo_distributionsad.jl b/test/inference/repgradelbo_distributionsad.jl
index 753e3cf3..fbe70ae9 100644
--- a/test/inference/repgradelbo_distributionsad.jl
+++ b/test/inference/repgradelbo_distributionsad.jl
@@ -6,7 +6,7 @@ AD_distributionsad = Dict(
 )
 
 if @isdefined(Mooncake)
-    AD_distributionsad[:Mooncake] = AutoMooncake(; config=Mooncake.config())
+    AD_distributionsad[:Mooncake] = AutoMooncake(; config=Mooncake.Config())
 end
 
 if @isdefined(Enzyme)
diff --git a/test/inference/repgradelbo_locationscale.jl b/test/inference/repgradelbo_locationscale.jl
index 4802f3d2..d1f0d7e4 100644
--- a/test/inference/repgradelbo_locationscale.jl
+++ b/test/inference/repgradelbo_locationscale.jl
@@ -6,7 +6,7 @@ AD_locationscale = Dict(
 )
 
 if @isdefined(Mooncake)
-    AD_locationscale[:Mooncake] = AutoMooncake(; config=Mooncake.config())
+    AD_locationscale[:Mooncake] = AutoMooncake(; config=Mooncake.Config())
 end
 
 if @isdefined(Enzyme)
diff --git a/test/inference/repgradelbo_locationscale_bijectors.jl b/test/inference/repgradelbo_locationscale_bijectors.jl
index 3135501e..e2a69d62 100644
--- a/test/inference/repgradelbo_locationscale_bijectors.jl
+++ b/test/inference/repgradelbo_locationscale_bijectors.jl
@@ -6,7 +6,7 @@ AD_locationscale_bijectors = Dict(
 )
 
 if @isdefined(Mooncake)
-    AD_locationscale_bijectors[:Mooncake] = AutoMooncake(; config=Mooncake.config())
+    AD_locationscale_bijectors[:Mooncake] = AutoMooncake(; config=Mooncake.Config())
 end
 
 if @isdefined(Enzyme)
diff --git a/test/inference/scoregradelbo_distributionsad.jl b/test/inference/scoregradelbo_distributionsad.jl
index 7903852a..9a621b40 100644
--- a/test/inference/scoregradelbo_distributionsad.jl
+++ b/test/inference/scoregradelbo_distributionsad.jl
@@ -6,7 +6,7 @@ AD_scoregradelbo_distributionsad = Dict(
 )
 
 if @isdefined(Mooncake)
-    AD_scoregradelbo_distributionsad[:Mooncake] = AutoMooncake(; config=Mooncake.config())
+    AD_scoregradelbo_distributionsad[:Mooncake] = AutoMooncake(; config=Mooncake.Config())
 end
 
 #if @isdefined(Enzyme)
diff --git a/test/inference/scoregradelbo_locationscale.jl b/test/inference/scoregradelbo_locationscale.jl
index 905542a6..753999de 100644
--- a/test/inference/scoregradelbo_locationscale.jl
+++ b/test/inference/scoregradelbo_locationscale.jl
@@ -6,7 +6,7 @@ AD_scoregradelbo_locationscale = Dict(
 )
 
 if @isdefined(Mooncake)
-    AD_scoregradelbo_locationscale[:Mooncake] = AutoMooncake(; config=Mooncake.config())
+    AD_scoregradelbo_locationscale[:Mooncake] = AutoMooncake(; config=Mooncake.Config())
 end
 
 if @isdefined(Enzyme)
diff --git a/test/interface/ad.jl b/test/interface/ad.jl
index 0be749f1..e23aec58 100644
--- a/test/interface/ad.jl
+++ b/test/interface/ad.jl
@@ -8,7 +8,7 @@ const interface_ad_backends = Dict(
 )
 
 if @isdefined(Mooncake)
-    interface_ad_backends[:Mooncake] = AutoMooncake(; config=Mooncake.config())
+    interface_ad_backends[:Mooncake] = AutoMooncake(; config=Mooncake.Config())
 end
 
 if @isdefined(Enzyme)
diff --git a/test/interface/repgradelbo.jl b/test/interface/repgradelbo.jl
index 3614dedf..be835e20 100644
--- a/test/interface/repgradelbo.jl
+++ b/test/interface/repgradelbo.jl
@@ -38,7 +38,7 @@ end
         ADTypes.AutoForwardDiff(), ADTypes.AutoReverseDiff(), ADTypes.AutoZygote()
     ]
     if @isdefined(Mooncake)
-        push!(ad_backends, AutoMooncake(; config=Mooncake.config()))
+        push!(ad_backends, AutoMooncake(; config=Mooncake.Config()))
     end
     if @isdefined(Enzyme)
         push!(