mtrl.agent.ds package

Submodules

mtrl.agent.ds.mt_obs module

Datastructure to wrap environment observation, task observation and other task-related information.

class mtrl.agent.ds.mt_obs.MTObs(env_obs: torch.Tensor, task_obs: Optional[torch.Tensor], task_info: Optional[mtrl.agent.ds.task_info.TaskInfo])

Bases: object

Class to wrap environment observation, task observation and other task-related information.

env_obs

task_info

task_obs

mtrl.agent.ds.task_info module

Datastructure to encapsulate the task-related information.

class mtrl.agent.ds.task_info.TaskInfo(encoding: Union[torch.Tensor, NoneType], compute_grad: bool, env_index: torch.Tensor)

Bases: object

compute_grad

encoding

env_index

Module contents