Fix: Test JAX only when installed

from cpprb.PyReplayBuffer import NstepBuffer

is_win = sys.platform.startswith("win")

if not is_win:

has_jax: bool = False

try:

    import jax.numpy as jnp

    has_jax = True

except ImportError:

    pass

@unittest.skipIf(is_win, "JAX doesn't support Windows")

@unittest.skipUnless(has_jax, "JAX is not installed")

class TestJAX(unittest.TestCase):

    def test_nstep_buffer(self):

        buffer = NstepBuffer({"obs": {}, "rew": {},  "done": {}, "next_obs": {}},