Remove Tracker

Author: penelopeysm

src/Turing.jl

@@ -106,7 +106,6 @@ export @model,                 # modelling
     AutoForwardDiff,        # ADTypes
     AutoReverseDiff,
     AutoZygote,
-    AutoTracker,
     AutoMooncake,
     setprogress!,           # debugging
     Flat,

src/essential/Essential.jl

@@ -12,7 +12,7 @@ using AdvancedVI
 using StatsFuns: logsumexp, softmax
 @reexport using DynamicPPL
 using ADTypes:
-    ADTypes, AutoForwardDiff, AutoTracker, AutoReverseDiff, AutoZygote, AutoMooncake
+    ADTypes, AutoForwardDiff, AutoReverseDiff, AutoZygote, AutoMooncake
 
 using AdvancedPS: AdvancedPS
 
@@ -21,7 +21,6 @@ include("container.jl")
 export @model,
     @varname,
     AutoForwardDiff,
-    AutoTracker,
     AutoZygote,
     AutoReverseDiff,
     AutoMooncake,

test/Project.toml

@@ -33,7 +33,6 @@ StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
 StatsFuns = "4c63d2b9-4356-54db-8cca-17b64c39e42c"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 TimerOutputs = "a759f4b9-e2f1-59dc-863e-4aeb61b1ea8f"
-Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [compat]
@@ -70,6 +69,5 @@ StableRNGs = "1"
 StatsBase = "0.33, 0.34"
 StatsFuns = "0.9.5, 1"
 TimerOutputs = "0.5"
-Tracker = "0.2.11"
 Zygote = "0.5.4, 0.6"
 julia = "1.3"

test/essential/ad.jl

@@ -88,20 +88,6 @@ end
         )
         x = map(x -> Float64(x), vi[SampleFromPrior()])
 
-        trackerℓ = LogDensityProblemsAD.ADgradient(Turing.AutoTracker(), ℓ)
-        if isdefined(Base, :get_extension)
-            @test trackerℓ isa
-                Base.get_extension(
-                LogDensityProblemsAD, :LogDensityProblemsADTrackerExt
-            ).TrackerGradientLogDensity
-        else
-            @test trackerℓ isa
-                LogDensityProblemsAD.LogDensityProblemsADTrackerExt.TrackerGradientLogDensity
-        end
-        @test trackerℓ.ℓ === ℓ
-        ∇E1 = LogDensityProblems.logdensity_and_gradient(trackerℓ, x)[2]
-        @test sort(∇E1) ≈ grad_FWAD atol = 1e-9
-
         zygoteℓ = LogDensityProblemsAD.ADgradient(Turing.AutoZygote(), ℓ)
         if isdefined(Base, :get_extension)
             @test zygoteℓ isa
@@ -149,7 +135,7 @@ end
 
         test_model_ad(wishart_ad(), logp3, [:v])
     end
-    @testset "Simplex Tracker, Zygote and ReverseDiff (with and without caching) AD" begin
+    @testset "Simplex Zygote and ReverseDiff (with and without caching) AD" begin
         @model function dir()
             return theta ~ Dirichlet(1 ./ fill(4, 4))
         end

test/ext/OptimInterface.jl

@@ -112,7 +112,8 @@ using Turing
         @test isapprox(map1.values.array, map2.values.array)
     end
 
-    # FIXME: Some models doesn't work for Tracker and ReverseDiff.
+    # FIXME: Some models don't work for ReverseDiff.
+    # TODO: Check if above statement is still correct
     @testset "MAP for $(model.f)" for model in DynamicPPL.TestUtils.DEMO_MODELS
         result_true = DynamicPPL.TestUtils.posterior_optima(model)
 

test/mcmc/Inference.jl

@@ -382,7 +382,6 @@ using Turing
         chn = sample(gdemo_default, alg, 1000)
     end
     @testset "vectorization @." begin
-        # https://github.com/FluxML/Tracker.jl/issues/119
         @model function vdemo1(x)
             s ~ InverseGamma(2, 3)
             m ~ Normal(0, sqrt(s))

test/optimisation/Optimisation.jl

@@ -1,7 +1,7 @@
 module OptimisationTests
 
 using ..Models: gdemo, gdemo_default
-using ..ADUtils: ADTypeCheckContext
+using ..ADUtils: ADUtils
 using Distributions
 using Distributions.FillArrays: Zeros
 using DynamicPPL: DynamicPPL
@@ -617,16 +617,14 @@ using Turing
         @assert get(result, :c) == (; :c => Array{Float64}[])
     end
 
-    @testset "ADType" begin
+    @testset "ADType" for adbackend in ADUtils.adbackends
         Random.seed!(222)
-        for adbackend in (AutoReverseDiff(), AutoForwardDiff(), AutoTracker())
-            m = DynamicPPL.contextualize(
-                gdemo_default, ADTypeCheckContext(adbackend, gdemo_default.context)
-            )
-            # These will error if the adbackend being used is not the one set.
-            maximum_likelihood(m; adtype=adbackend)
-            maximum_a_posteriori(m; adtype=adbackend)
-        end
+        m = DynamicPPL.contextualize(
+            gdemo_default, ADUtils.ADTypeCheckContext(adbackend, gdemo_default.context)
+        )
+        # These will error if the adbackend being used is not the one set.
+        maximum_likelihood(m; adtype=adbackend)
+        maximum_a_posteriori(m; adtype=adbackend)
     end
 end
 

test/test_utils/ad_utils.jl

@@ -6,7 +6,6 @@ using Random: Random
 using ReverseDiff: ReverseDiff
 using Mooncake: Mooncake
 using Test: Test
-using Tracker: Tracker
 using Turing: Turing
 using Turing: DynamicPPL
 using Zygote: Zygote
@@ -31,19 +30,10 @@ const eltypes_by_adtype = Dict(
         ReverseDiff.TrackedVecOrMat,
         ReverseDiff.TrackedVector,
     ),
+    Turing.AutoMooncake => (Mooncake.CoDual,),
     # Zygote.Dual is actually the same as ForwardDiff.Dual, so can't distinguish between the
     # two by element type. However, we have other checks for Zygote, see check_adtype.
     Turing.AutoZygote => (Zygote.Dual,),
-    Turing.AutoTracker => (
-        Tracker.Tracked,
-        Tracker.TrackedArray,
-        Tracker.TrackedMatrix,
-        Tracker.TrackedReal,
-        Tracker.TrackedStyle,
-        Tracker.TrackedVecOrMat,
-        Tracker.TrackedVector,
-    ),
-    Turing.AutoMooncake => (Mooncake.CoDual,),
 )
 
 """
@@ -247,7 +237,8 @@ Test.@testset "ADTypeCheckContext" begin
         Turing.AutoForwardDiff(),
         Turing.AutoReverseDiff(),
         Turing.AutoZygote(),
-        Turing.AutoTracker(),
+        # TODO: Mooncake
+        # Turing.AutoMooncake(config=nothing),
     )
     for actual_adtype in adtypes
         sampler = Turing.HMC(0.1, 5; adtype=actual_adtype)
@@ -285,7 +276,7 @@ All the ADTypes on which we want to run the tests.
 adbackends = [
     Turing.AutoForwardDiff(; chunksize=0),
     Turing.AutoReverseDiff(; compile=false),
-    Turing.AutoMooncake(; config=Mooncake.Config()),
+    Turing.AutoMooncake(; config=nothing),
 ]
 
 end