no short func

impICNF · Jan 23, 2025 · ab11209 · ab11209
1 parent a5dc936
commit ab11209
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 6 deletions.
diff --git a/src/exts/mlj_ext/core_cond_icnf.jl b/src/exts/mlj_ext/core_cond_icnf.jl
@@ -108,7 +108,9 @@ function MLJModelInterface.transform(model::CondICNFModel, fitresult, XYnew)
 
     tst = @timed if model.m.compute_mode isa VectorMode
         logp̂x = broadcast(
-            (x, y) -> first(inference(model.m, TestMode(), x, y, ps, st)),
+            function (x, y)
+                first(inference(model.m, TestMode(), x, y, ps, st))
+            end,
             eachcol(xnew),
             eachcol(ynew),
         )

diff --git a/src/exts/mlj_ext/core_icnf.jl b/src/exts/mlj_ext/core_icnf.jl
@@ -102,7 +102,9 @@ function MLJModelInterface.transform(model::ICNFModel, fitresult, Xnew)
     (ps, st) = fitresult
 
     tst = @timed if model.m.compute_mode isa VectorMode
-        logp̂x = broadcast(x -> first(inference(model.m, TestMode(), x, ps, st)), eachcol(xnew))
+        logp̂x = broadcast(function (x)
+                first(inference(model.m, TestMode(), x, ps, st))
+            end, eachcol(xnew))
     elseif model.m.compute_mode isa MatrixMode
         logp̂x = first(inference(model.m, TestMode(), xnew, ps, st))
     else

diff --git a/test/call_tests.jl b/test/call_tests.jl
@@ -190,8 +190,12 @@ Test.@testset "Call Tests" begin
             )
             Test.@test !isnothing(icnf((r, r2), ps, st))
 
-            diff_loss = x -> ContinuousNormalizingFlows.loss(icnf, omode, r, r2, x, st)
-            diff2_loss = x -> ContinuousNormalizingFlows.loss(icnf, omode, x, r2, ps, st)
+            diff_loss = function (x)
+                ContinuousNormalizingFlows.loss(icnf, omode, r, r2, x, st)
+            end
+            diff2_loss = function (x)
+                ContinuousNormalizingFlows.loss(icnf, omode, x, r2, ps, st)
+            end
         else
             Test.@test !isnothing(
                 ContinuousNormalizingFlows.inference(icnf, omode, r, ps, st),
@@ -209,8 +213,12 @@ Test.@testset "Call Tests" begin
             Test.@test !isnothing(ContinuousNormalizingFlows.loss(icnf, omode, r, ps, st))
             Test.@test !isnothing(icnf(r, ps, st))
 
-            diff_loss = x -> ContinuousNormalizingFlows.loss(icnf, omode, r, x, st)
-            diff2_loss = x -> ContinuousNormalizingFlows.loss(icnf, omode, x, ps, st)
+            diff_loss = function (x)
+                ContinuousNormalizingFlows.loss(icnf, omode, r, x, st)
+            end
+            diff2_loss = function (x)
+                ContinuousNormalizingFlows.loss(icnf, omode, x, ps, st)
+            end
         end
 
         if mt <: Union{