fixing compat and test with mooncake

Author: zuhengxu

Project.toml

@@ -25,18 +25,18 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [compat]
 ADTypes = "0.1, 0.2, 1"
-Bijectors = "0.12.6, 0.13, 0.14"
+Bijectors = "0.12.6, 0.13, 0.14, 0.15"
 DifferentiationInterface = "0.6"
 Distributions = "0.25"
 DocStringExtensions = "0.9"
 Mooncake = "0.4.95"
 ForwardDiff = "0.10.25"
-Optimisers = "0.2.16, 0.3"
+Optimisers = "0.2.16, 0.3, 0.4"
 ProgressMeter = "1.0.0"
 Requires = "1"
 ReverseDiff = "1.14"
 StatsBase = "0.33, 0.34"
-Zygote = "0.6"
+Zygote = "0.6, 0.7"
 julia = "1.6"
 
 [extras]

test/ad.jl

@@ -12,7 +12,7 @@
             ADTypes.AutoForwardDiff(; chunksize=chunksize),
             ADTypes.AutoForwardDiff(),
             ADTypes.AutoReverseDiff(false),
-            ADTypes.AutoMooncake(; config=ADTypes.Mooncake.Config()),
+            ADTypes.AutoMooncake(; config=Mooncake.Config()),
         ]
             value, grad = NormalizingFlows._value_and_gradient(f, at, x, y, z)
             @test DiffResults.value(out) ≈ f(x, y, z)
@@ -26,7 +26,7 @@ end
         ADTypes.AutoZygote(),
         ADTypes.AutoForwardDiff(),
         ADTypes.AutoReverseDiff(false),
-        ADTypes.AutoMooncake(; config=ADTypes.Mooncake.Config()),
+        ADTypes.AutoMooncake(; config=Mooncake.Config()),
     ]
         @testset "$T" for T in [Float32, Float64]
             μ = 10 * ones(T, 2)

test/interface.jl

@@ -5,7 +5,7 @@
         ADTypes.AutoForwardDiff(; chunksize=chunksize),
         ADTypes.AutoForwardDiff(),
         ADTypes.AutoReverseDiff(),
-        ADTypes.AutoMooncake(; config = ADTypes.Mooncake.Config()),
+        ADTypes.AutoMooncake(; config = Mooncake.Config()),
     ]
         @testset "$T" for T in [Float32, Float64]
             μ = 10 * ones(T, 2)