fix jet tests

Author: prbzrg

examples/usage.jl

@@ -83,7 +83,7 @@ display(res_df)
 # Plot It
 using CairoMakie
 f = Figure()
-ax = Makie.Axis(f[1, 1]; title = "Result")
+ax = Axis(f[1, 1]; title = "Result")
 lines!(ax, 0.0f0 .. 1.0f0, x -> pdf(data_dist, x); label = "actual")
 lines!(ax, 0.0f0 .. 1.0f0, x -> pdf(d, vcat(x)); label = "estimated")
 axislegend(ax)

test/quality_tests/checkby_JET_tests.jl

@@ -1,8 +1,7 @@
 Test.@testset "CheckByJET" begin
     JET.test_package(
         ContinuousNormalizingFlows;
-        target_modules = [ContinuousNormalizingFlows],
-        mode = :typo,
+        target_modules = (ContinuousNormalizingFlows,),
     )
 
     mts = Type{<:ContinuousNormalizingFlows.AbstractICNF}[ContinuousNormalizingFlows.ICNF]
@@ -118,29 +117,37 @@ Test.@testset "CheckByJET" begin
 
         if cond
             ContinuousNormalizingFlows.loss(icnf, omode, r, r2, ps, st)
-            JET.test_call(
-                ContinuousNormalizingFlows.loss,
-                Base.typesof(icnf, omode, r, r2, ps, st);
-                target_modules = [ContinuousNormalizingFlows],
-                mode = :typo,
+            JET.@test_call target_modules = (ContinuousNormalizingFlows,) ContinuousNormalizingFlows.loss(
+                icnf,
+                omode,
+                r,
+                r2,
+                ps,
+                st,
             )
-            JET.test_opt(
-                ContinuousNormalizingFlows.loss,
-                Base.typesof(icnf, omode, r, r2, ps, st);
-                target_modules = [ContinuousNormalizingFlows],
+            JET.@test_opt target_modules = (ContinuousNormalizingFlows,) ContinuousNormalizingFlows.loss(
+                icnf,
+                omode,
+                r,
+                r2,
+                ps,
+                st,
             )
         else
             ContinuousNormalizingFlows.loss(icnf, omode, r, ps, st)
-            JET.test_call(
-                ContinuousNormalizingFlows.loss,
-                Base.typesof(icnf, omode, r, ps, st);
-                target_modules = [ContinuousNormalizingFlows],
-                mode = :typo,
+            JET.@test_call target_modules = (ContinuousNormalizingFlows,) ContinuousNormalizingFlows.loss(
+                icnf,
+                omode,
+                r,
+                ps,
+                st,
             )
-            JET.test_opt(
-                ContinuousNormalizingFlows.loss,
-                Base.typesof(icnf, omode, r, ps, st);
-                target_modules = [ContinuousNormalizingFlows],
+            JET.@test_opt target_modules = (ContinuousNormalizingFlows,) ContinuousNormalizingFlows.loss(
+                icnf,
+                omode,
+                r,
+                ps,
+                st,
             )
         end
     end