added metadata and reward_range for each tasks

This commit is contained in:
NM512
2023-07-23 21:55:22 +09:00
parent 106317015d
commit 2c933da684
3 changed files with 6 additions and 1 deletions

View File

@@ -3,12 +3,14 @@ import numpy as np
class Crafter:
metadata = {}
def __init__(self, task, size=(64, 64), seed=None):
assert task in ("reward", "noreward")
import crafter
self._env = crafter.Env(size=size, reward=(task == "reward"), seed=seed)
self._achievements = crafter.constants.achievements.copy()
self.reward_range = [-np.inf, np.inf]
@property
def observation_space(self):