From 02851c8bbe6b62ad779244368360844afdb7fa18 Mon Sep 17 00:00:00 2001 From: Shravan Goswami Date: Mon, 10 Mar 2025 21:36:42 +0530 Subject: [PATCH] removed Zygote & Tracker test --- test/Project.toml | 2 -- test/runtests.jl | 1 - test/test_util.jl | 16 ---------------- 3 files changed, 19 deletions(-) diff --git a/test/Project.toml b/test/Project.toml index e0fbbb8c5..c8aa6390e 100644 --- a/test/Project.toml +++ b/test/Project.toml @@ -26,7 +26,6 @@ ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267" Serialization = "9e88b42a-f829-5b0c-bbe9-9e923198166b" StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3" Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40" -Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c" Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f" [compat] @@ -51,6 +50,5 @@ Mooncake = "0.4.95" OrderedCollections = "1" ReverseDiff = "1" StableRNGs = "1" -Tracker = "0.2.23" Zygote = "0.6" julia = "1.10" diff --git a/test/runtests.jl b/test/runtests.jl index caddef5f9..02c999caf 100644 --- a/test/runtests.jl +++ b/test/runtests.jl @@ -14,7 +14,6 @@ using MacroTools using MCMCChains using Mooncake: Mooncake using StableRNGs -using Tracker using ReverseDiff using Zygote using Compat diff --git a/test/test_util.jl b/test/test_util.jl index d831a5ea6..69f9f7656 100644 --- a/test/test_util.jl +++ b/test/test_util.jl @@ -23,24 +23,8 @@ function test_model_ad(model, logp_manual) # Gradients based on the manual implementation. grad = ForwardDiff.gradient(logp_manual, x) - y, back = Tracker.forward(logp_manual, x) - @test Tracker.data(y) ≈ lp - @test Tracker.data(back(1)[1]) ≈ grad - - y, back = Zygote.pullback(logp_manual, x) - @test y ≈ lp - @test back(1)[1] ≈ grad - # Gradients based on the model. @test ForwardDiff.gradient(logp_model, x) ≈ grad - - y, back = Tracker.forward(logp_model, x) - @test Tracker.data(y) ≈ lp - @test Tracker.data(back(1)[1]) ≈ grad - - y, back = Zygote.pullback(logp_model, x) - @test y ≈ lp - @test back(1)[1] ≈ grad end """