self.normalize = normalize
self.unwrap_time_limit = unwrap_time_limit
self._Serializable__initialize(locals())
super(DmControlAdapter, self).__init__(domain, task, *args, **kwargs)
if env is None:
assert (domain is not None and task is not None), (domain, task)
env = suite.load(
After Change
)
self._env_kwargs = kwargs
else:
assert not kwargs
assert domain is None and task is None, (domain, task)
assert isinstance(env.observation_spec(), OrderedDict)
self.observation_keys = (