reno.pymc.resolve_pt_scan_args

reno.pymc.resolve_pt_scan_args(model, args)

Take the set of arguments passed by pytensor to a target scan function, and turn it into a dictionary for passing into reno math components’ pt() calls.

(This is necessary to ensure the returned pytensor symbolic operations are referring to the correct tensorvariables passed into scan.)

Parameters:

• model (Model)

• args (list[TensorVariable])

Return type:

dict[str, TensorVariable]