Skip to content

Commit 080366a

Browse files
committed
Fix JET test
1 parent 727da63 commit 080366a

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

test/ext/DynamicPPLJETExt.jl

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,14 +66,15 @@
6666
varinfo = DynamicPPL.Experimental.determine_suitable_varinfo(model)
6767
# They should all result in typed.
6868
@test varinfo isa DynamicPPL.TypedVarInfo
69-
# But let's also make sure that they're not lying.
69+
# But let's also make sure that they're not lying by checking the
70+
# type inference with a typed varinfo.
7071
f_eval, argtypes_eval = DynamicPPL.DebugUtils.gen_evaluator_call_with_types(
71-
model, varinfo
72+
model, VarInfo(model)
7273
)
7374
JET.test_call(f_eval, argtypes_eval)
7475

7576
f_sample, argtypes_sample = DynamicPPL.DebugUtils.gen_evaluator_call_with_types(
76-
model, varinfo, DynamicPPL.SamplingContext()
77+
model, VarInfo(model), DynamicPPL.SamplingContext()
7778
)
7879
JET.test_call(f_sample, argtypes_sample)
7980
end

0 commit comments

Comments
 (0)