diff --git a/mo_gymnasium/envs/minecart/__init__.py b/mo_gymnasium/envs/minecart/__init__.py index 349bae51..8be0f988 100644 --- a/mo_gymnasium/envs/minecart/__init__.py +++ b/mo_gymnasium/envs/minecart/__init__.py @@ -9,6 +9,13 @@ max_episode_steps=1000, ) +register( + id="minecart-rgb-v0", + entry_point="mo_gymnasium.envs.minecart.minecart:Minecart", + kwargs={"image_observation": True}, + max_episode_steps=1000, +) + register( id="minecart-deterministic-v0", entry_point="mo_gymnasium.envs.minecart.minecart:Minecart", diff --git a/mo_gymnasium/envs/minecart/minecart.py b/mo_gymnasium/envs/minecart/minecart.py index 568a14f9..846231a3 100644 --- a/mo_gymnasium/envs/minecart/minecart.py +++ b/mo_gymnasium/envs/minecart/minecart.py @@ -1,7 +1,6 @@ import itertools import json import math -from copy import deepcopy from math import ceil from pathlib import Path from typing import List, Optional @@ -566,7 +565,7 @@ def reset(self, seed=None, **kwargs): super().reset(seed=seed) if self.screen is None and self.image_observation: - self.render(mode="rgb_array") # init pygame + self.render() # init pygame if self.image_observation: self.render_pygame() @@ -675,13 +674,6 @@ def close(self): pygame.display.quit() pygame.quit() - def __deepcopy__(self, memo): - this_copy = Minecart(self.image_observation, self.config) - this_copy.cart = deepcopy(self.cart) - this_copy.mines = deepcopy(self.mines) - this_copy.end = self.end - return this_copy - class Mine: """Class representing an individual Mine"""