From 71d03bd7a46be9f4f9a2f9305034d537bfb77b12 Mon Sep 17 00:00:00 2001
From: Guillaume Dalle <22795598+gdalle@users.noreply.github.com>
Date: Wed, 5 Jul 2023 10:37:00 +0200
Subject: [PATCH] Tmp fix for extension-related JET error

---
 test/code.jl | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/test/code.jl b/test/code.jl
index 33899a3..eeb9ecd 100644
--- a/test/code.jl
+++ b/test/code.jl
@@ -5,11 +5,14 @@
     Aqua.test_ambiguities(InferOpt; exclude=[StatsBase.TestStat])
 end
 
-@testitem "Correctness (JET.jl)" begin
+@testitem "Correctness (JET.jl)" default_imports = false begin
     using JET
     using DifferentiableFrankWolfe
+    using InferOpt
     if VERSION >= v"1.9"
-        @test_skip JET.test_package(InferOpt; target_defined_modules=true)
+        JET.test_package(InferOpt; target_modules=[InferOpt])
+        # TODO: why does the following line fail even though the method is defined in the extension?
+        # JET.test_package(InferOpt, target_defined_modules=true)
     end
 end