Make tests compatible with newer version of JAX#133
Merged
michaelosthege merged 1 commit intopymc-devs:mainfrom Dec 14, 2022
Merged
Make tests compatible with newer version of JAX#133michaelosthege merged 1 commit intopymc-devs:mainfrom
michaelosthege merged 1 commit intopymc-devs:mainfrom
