pantheonrl.common.trajsaver.SimultaneousTransitions

class SimultaneousTransitions(egoobs, egoacts, altobs, altacts, flags)[source]

Bases: MultiTransitions

Class that stores transitions from SimultaneousEnv

Methods

get_alt_transitions

Returns the partner's transitions

get_ego_transitions

Returns the ego's transitions

read_transition

Construct SimultaneousTransitions from file

write_transition

Write transition to a given file.

Attributes

egoobs

egoacts

altobs

altacts

flags

Parameters:
  • egoobs (ndarray) –

  • egoacts (ndarray) –

  • altobs (ndarray) –

  • altacts (ndarray) –

  • flags (ndarray) –

get_alt_transitions()[source]

Returns the partner’s transitions

Return type:

TransitionsMinimal

get_ego_transitions()[source]

Returns the ego’s transitions

Return type:

TransitionsMinimal

classmethod read_transition(file, obs_space, act_space)[source]

Construct SimultaneousTransitions from file

write_transition(file)[source]

Write transition to a given file.