mirror of
https://github.com/gryf/coach.git
synced 2025-12-18 11:40:18 +01:00
9f625c197b
* fix for fetch rendering - removing code which was once required with older gym versions. images are now rendered correctly by default with the latest gym. * fixing mujoco camera id failure
A custom environment implementation should look like this:
from coach.filters.input_filter import InputFilter
class CustomFilter(InputFilter):
def __init__(self):
...
def _filter(self, env_response: EnvResponse) -> EnvResponse:
...
def _get_filtered_observation_space(self, input_observation_space: ObservationSpace) -> ObservationSpace:
...
def _get_filtered_reward_space(self, input_reward_space: RewardSpace) -> RewardSpace:
...
def _validate_input_observation_space(self, input_observation_space: ObservationSpace):
...
def _reset(self):
...