pantheonrl.common.trajsaver.TurnBasedTransitions

class TurnBasedTransitions(obs, acts, flags)[source]

Bases: MultiTransitions

Class that stores transitions from TurnBasedEnv

Methods

get_alt_transitions

Returns the partner's transitions

get_ego_transitions

Returns the ego's transitions

read_transition

Construct TurnBasedTransitions from file

write_transition

Write transition to a given file.

Attributes

obs

acts

flags

Parameters:
  • obs (ndarray) –

  • acts (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 TurnBasedTransitions from file

write_transition(file)[source]

Write transition to a given file.