typed_varinfo and untyped_varinfo handles wrapping passed context

in sampling context now so no need to handle this explicitly elsewhere

Author: torfjelde

ext/DynamicPPLJETExt.jl

@@ -51,14 +51,8 @@ end
 function DynamicPPL._determine_varinfo_jet(
     model::DynamicPPL.Model, context::DynamicPPL.AbstractContext; only_tilde::Bool=true
 )
-    # We need a sampling context in the stack to initialize the varinfo.
-    sampling_context = if DynamicPPL.hassampler(context)
-        context
-    else
-        DynamicPPL.typed_varinfo(model, DynamicPPL.SamplingContext(context))
-    end
     # First we try with the typed varinfo.
-    varinfo = DynamicPPL.typed_varinfo(model, sampling_context)
+    varinfo = DynamicPPL.typed_varinfo(model, context)
     issuccess = true
 
     # Let's make sure that both evaluation and sampling doesn't result in type errors.
@@ -78,7 +72,7 @@ function DynamicPPL._determine_varinfo_jet(
     else
         # Warn the user that we can't use the type stable one.
         @warn "Model seems incompatible with typed varinfo. Falling back to untyped varinfo."
-        DynamicPPL.untyped_varinfo(model, sampling_context)
+        DynamicPPL.untyped_varinfo(model, context)
     end
 end