Coverage for adaro_rl / zoo / Template / agent.py: 100%

1repo_id = None 

2filename = None 

3 

4 

5agent_config = { 

6 "algo": None, # stable-baselines3 algorithm class (tested yet with stable_baslines3.PPO and stable_baslines3.SAC) 

7 "algo_kwargs": {}, # stable-baselines3 algorithm kwargs to train the base agent 

8} 

9training_kwargs = {"total_timesteps": None, "n_eval_episodes": None, "eval_freq": None} 

10 

11 

12adversary_config = {"algo": None, "algo_kwargs": {}} 

13adversary_training_kwargs = { 

14 "total_timesteps": None, 

15 "n_eval_episodes": None, 

16 "eval_freq": None, 

17} 

18 

19 

20finetuned_agent_config = {"algo": None, "algo_kwargs": {}} 

21finetuned_training_kwargs = { 

22 "total_timesteps": None, 

23 "n_eval_episodes": None, 

24 "eval_freq": None, 

25} 

26 

27 

28n_eval_episodes = None # number of episodes to evaluate the agent