Wrappers

Labels extractor wrapper.

class gym_saturation.wrappers.labels_extractor.LabelsExtractor(env: Env[ObsType, ActType])

Labels extractor wrapper.

>>> from gym_saturation.envs.vampire_env import VampireEnv
>>> env = LabelsExtractor(VampireEnv())
>>> observation, info = env.reset()
>>> type(observation)
<class 'dict'>
>>> observation.keys()
dict_keys(['labels', 'observation'])
>>> type(observation["labels"])
<class 'tuple'>
>>> type(observation["labels"][0])
<class 'str'>

observation(observation: tuple[str, ...]) → dict[str, tuple[str, ...]]

Return a modified observation.

Parameters:

observation – The observation

Returns:

The modified observation