Missing one-hot vector

Hello 😃

The documentation mentions that MT10 and MT50 augment environment observations with a one-hot vector which identifies the task.. When I create a MT10 instance (using the code below) I do not get the task id. Could you please explain what am I missing.

import metaworld
import random

mt10 = metaworld.MT10() # Construct the benchmark, sampling tasks

training_envs = []
for name, env_cls in mt10.train_classes.items():
  env = env_cls()
  task = random.choice([task for task in mt10.train_tasks
                        if task.env_name == name])
  env.set_task(task)
  training_envs.append(env)

for env in training_envs:
  obs = env.reset()  # Reset environment
  a = env.action_space.sample()  # Sample an action
  obs, reward, done, info = env.step(a)  # Step the environoment with the sampled random action

The shape of obs is (12,) and not (13,)

[avnishn]

Hi @shagunsodhani I’m going to close this issue for now because I believe that I have answered your question. Please do @ me and reopen this issue if you have more questions 😃

[shagunsodhani]

Thanks for all the replies @avnishn . I will get back if I have any other questions.