Skip to content
Snippets Groups Projects
Commit abbbed80 authored by Verena Kain's avatar Verena Kain
Browse files

removing gym also from tests

parent 2d429041
No related branches found
No related tags found
No related merge requests found
Pipeline #9981513 failed
......@@ -14,7 +14,7 @@ import typing as t
from unittest.mock import Mock
import cernml.coi
import gym.envs.registration
import gymnasium.envs.registration
import numpy as np
import pytest
from cernml.optimizers import make, registry
......@@ -47,7 +47,7 @@ def optimizable() -> cernml.coi.SingleOptimizable:
make_mock_constraint(shape=(3,)),
]
result.compute_single_objective.side_effect = np.linalg.norm
result.optimization_space = gym.spaces.Box(-1.0, 1.0, shape=(3,))
result.optimization_space = gymnasium.spaces.Box(-1.0, 1.0, shape=(3,))
result.get_initial_params.return_value = result.optimization_space.sample()
result.return_value = result
result.objective_name = ""
......@@ -56,7 +56,7 @@ def optimizable() -> cernml.coi.SingleOptimizable:
result.spec = Mock(
id=f"MockEnv-{id(result)}-v0",
entry_point=result,
spec=gym.envs.registration.EnvSpec,
spec=gymnasium.envs.registration.EnvSpec,
)
result.spec.make.return_value = result
return result
......
......@@ -14,7 +14,7 @@ from __future__ import annotations
import typing as t
from cernml.coi import Configurable, FunctionOptimizable, Problem, SingleOptimizable
from gym import Env
from gymnasium import Env
if t.TYPE_CHECKING: # pragma: no cover
import sys
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment