From 080366acf7bb62ece2277326c95f49ca7692fa9c Mon Sep 17 00:00:00 2001 From: Penelope Yong Date: Tue, 21 Jan 2025 18:54:25 +0000 Subject: [PATCH] Fix JET test --- test/ext/DynamicPPLJETExt.jl | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/test/ext/DynamicPPLJETExt.jl b/test/ext/DynamicPPLJETExt.jl index b95107b2d..104f35828 100644 --- a/test/ext/DynamicPPLJETExt.jl +++ b/test/ext/DynamicPPLJETExt.jl @@ -66,14 +66,15 @@ varinfo = DynamicPPL.Experimental.determine_suitable_varinfo(model) # They should all result in typed. @test varinfo isa DynamicPPL.TypedVarInfo - # But let's also make sure that they're not lying. + # But let's also make sure that they're not lying by checking the + # type inference with a typed varinfo. f_eval, argtypes_eval = DynamicPPL.DebugUtils.gen_evaluator_call_with_types( - model, varinfo + model, VarInfo(model) ) JET.test_call(f_eval, argtypes_eval) f_sample, argtypes_sample = DynamicPPL.DebugUtils.gen_evaluator_call_with_types( - model, varinfo, DynamicPPL.SamplingContext() + model, VarInfo(model), DynamicPPL.SamplingContext() ) JET.test_call(f_sample, argtypes_sample) end