From 2e3f0b5e99f23bb40ba5418d33e0831d65c9fec1 Mon Sep 17 00:00:00 2001 From: Maximilian Huettenrauch Date: Tue, 26 Mar 2024 14:26:27 +0100 Subject: [PATCH] move doc string --- examples/mujoco/mujoco_ppo_hl_multi.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/examples/mujoco/mujoco_ppo_hl_multi.py b/examples/mujoco/mujoco_ppo_hl_multi.py index 9911379..c5edb1f 100644 --- a/examples/mujoco/mujoco_ppo_hl_multi.py +++ b/examples/mujoco/mujoco_ppo_hl_multi.py @@ -1,7 +1,4 @@ #!/usr/bin/env python3 -"""Use the high-level API of TianShou to evaluate the PPO algorithm on a MuJoCo environment with multiple seeds for a -given configuration. After the agents are trained, the results are evaluated using rliable API. -""" import os from collections.abc import Sequence @@ -53,6 +50,10 @@ def main( norm_adv: bool = False, recompute_adv: bool = True, ) -> str: + """Use the high-level API of TianShou to evaluate the PPO algorithm on a MuJoCo environment with multiple seeds for + a given configuration. The results for each run are stored in separate sub-folders. After the agents are trained, + the results are evaluated using rliable API. + """ log_name = os.path.join("log", task, "ppo", datetime_tag()) experiment_config.persistence_base_dir = log_name