make JET and AllocCheck dependency optional

Author: lkdvos

test/Project.toml

@@ -2,15 +2,13 @@ name = "TensorKitTests"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
-AllocCheck = "9b6a8646-10ed-4001-bbdc-1d2f46dfbb1a"
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 ChainRulesTestUtils = "cdddcdb0-9152-4a09-a978-84456f9df70a"
 Combinatorics = "861a8166-3701-5b0c-9a16-15d98fcdc6aa"
 FiniteDifferences = "26cc04aa-876d-5657-8c51-4c34ba976000"
 GPUArrays = "0c68f7d7-f131-5f86-a1c3-88cf8149b2d7"
-JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MatrixAlgebraKit = "6c742aac-3347-4629-af66-fc926824e5e4"
 Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
@@ -31,11 +29,9 @@ TensorKit = {path = ".."}
 
 [compat]
 Aqua = "0.6, 0.7, 0.8"
-AllocCheck = "0.2"
 ChainRulesTestUtils = "1"
 Combinatorics = "1"
 GPUArrays = "11.3.1"
-JET = "0.9, 0.10, 0.11"
 ParallelTestRunner = "2"
 Test = "1"
 TestExtras = "0.2,0.3"

test/mooncake/tangent.jl

@@ -2,9 +2,23 @@ using Test, TestExtras
 using TensorKit
 using Mooncake
 using Random
-using JET, AllocCheck
 
-using .TestSetup: _repartition
+# Since we aren't actively depending on JET and AllocCheck, but only implicitly through the package extensions
+# of Mooncake, we don't want to break nightly tests etc. If this fails, tests still run but without the JET tests.
+const JET_AVAILABLE = let
+    try
+        import Pkg
+        current = Base.active_project()
+        Pkg.activate(; temp = true, io = devnull)
+        Pkg.add(["JET", "AllocCheck"]; io = devnull)
+        @eval using JET, AllocCheck
+        Pkg.activate(current; io = devnull)
+        true
+    catch
+        @warn "Cannot load JET and/or AllocCheck"
+        false
+    end
+end
 
 mode = Mooncake.ReverseMode
 rng = Random.default_rng()