gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-01 14:10:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update the jumpy error to specify the pip install is jax-jumpy (#255)

Browse Source
This commit is contained in:
Mark Towers
2023-01-11 13:38:32 +00:00
committed by GitHub
parent c52ef43a1b
commit febe5265c4
3 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
gymnasium/experimental/wrappers/lambda_action.py
View File

@@ -8,7 +8,11 @@ from __future__ import annotations
from typing import Callable from typing import Callable
import jumpy as jp
try:
import jumpy as jp
except ImportError as e:
raise ImportError("Jumpy is not installed, run `pip install jax-jumpy`") from e
import numpy as np import numpy as np
import gymnasium as gym import gymnasium as gym

6
gymnasium/experimental/wrappers/lambda_observations.py
View File

@@ -16,7 +16,11 @@ from __future__ import annotations
from typing import Any, Callable, Sequence from typing import Any, Callable, Sequence
from typing_extensions import Final from typing_extensions import Final
import jumpy as jp
try:
import jumpy as jp
except ImportError as e:
raise ImportError("Jumpy is not installed, run `pip install jax-jumpy`") from e
import numpy as np import numpy as np
import gymnasium as gym import gymnasium as gym

6
gymnasium/experimental/wrappers/stateful_observation.py
View File

@@ -11,7 +11,11 @@ from collections import deque
from typing import Any, SupportsFloat from typing import Any, SupportsFloat
from typing_extensions import Final from typing_extensions import Final
import jumpy as jp
try:
import jumpy as jp
except ImportError as e:
raise ImportError("Jumpy is not installed, run `pip install jax-jumpy`") from e
import numpy as np import numpy as np
import gymnasium as gym import gymnasium as gym