gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-20 22:12:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Skip Jax and PyTorch tests if module is missing (#466)

Browse Source
This commit is contained in:
ChristofKaufmann
2023-04-25 03:47:51 -07:00
committed by GitHub
parent fa5fdb435e
commit 4350e9c6b9
7 changed files with 45 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/experimental/wrappers/test_jax_to_numpy.py
View File

@@ -1,16 +1,18 @@
"""Test suite for JaxToNumpyV0."""
import jax.numpy as jnp
import numpy as np
import pytest
from gymnasium.experimental.wrappers.jax_to_numpy import (
jnp = pytest.importorskip("jax.numpy")
from gymnasium.experimental.wrappers.jax_to_numpy import ( # noqa: E402
JaxToNumpyV0,
jax_to_numpy,
numpy_to_jax,
)
from gymnasium.utils.env_checker import data_equivalence
from tests.testing_env import GenericTestEnv
from gymnasium.utils.env_checker import data_equivalence # noqa: E402
from tests.testing_env import GenericTestEnv # noqa: E402
@pytest.mark.parametrize(