gridmind.wrappers.policy_wrappers package

Submodules

gridmind.wrappers.policy_wrappers.base_policy_wrapper module

class gridmind.wrappers.policy_wrappers.base_policy_wrapper.BasePolicyWrapper(policy: BasePolicy)[source]

Bases: ABC

get_action(state)[source]

get_action_prob(state, action)[source]

get_policy()[source]

update(state, action)[source]

gridmind.wrappers.policy_wrappers.preprocessed_observation_policy_wrapper module

class gridmind.wrappers.policy_wrappers.preprocessed_observation_policy_wrapper.PreprocessedObservationPolicyWrapper(policy, preprocess_fn)[source]

Bases: BasePolicyWrapper

get_action(state)[source]

get_action_prob(state, action)[source]

update(state, action)[source]

Module contents