I followed this example with my own data (2 input, 1 continuous output), using v4.1.4 and sampled the trace with jax.
with model:
ppc = pm.sample_posterior_predictive(trace, samples=100, var_names=["fnew"])
returns
IncorrectArgumentsError Traceback (most recent call last)
<ipython-input-20-7da9f5d1c2b9> in <module>
3
4 with model:
----> 5 ppc = pm.sample_posterior_predictive(trace, samples=100, var_names=["fnew"])
/usr/local/lib/python3.7/dist-packages/pymc/sampling.py in sample_posterior_predictive(trace, samples, model, var_names, keep_size, random_seed, progressbar, return_inferencedata, extend_inferencedata, predictions, idata_kwargs, compile_kwargs)
1857 if keep_size and samples is not None:
1858 raise IncorrectArgumentsError(
-> 1859 "Should not specify both keep_size and samples arguments. "
1860 "See the docstring of the samples argument for more details."
1861 )