rlberry.utils.check_experiment_manager

rlberry.utils.check_experiment_manager(agent, env='continuous_state', init_kwargs=None)

Check that the agent is compatible with ExperimentManager.

Parameters:

agent: rlberry agent module

Agent class to test.

env: tuple (env_ctor, env_kwargs) or str in {“continuous_state”, “discrete_state”}, default=”continuous_state”

if tuple, env is the constructor and keywords of the env on which to test. if str in {“continuous_state”, “discrete_state”}, we use a default Benchmark environment.

init_kwargsdict

Arguments required by the agent’s constructor.