Unit Test Fails (5): test_model_basic_renewal.py

[akruel01]

Still uses JAX, disconnect with scripts that have been converted (distutil.py)

Line 30: gen_int = DeterministicPMF(
            jnp.array([0.25, 0.25, 0.25, 0.25]), name="gen_int"
        )
pyrenew/distutil.py:50 > if not torch.all(discrete_dist >= 0):
TypeError: all() received an invalid combination of arguments - got (jaxlib.xla_extension.ArrayImpl), but expected ... [bunch of tensor options]

etc. jax =/= torch errors