From ab704ec1c487a58fce2edc68fd97fb9a0e4f36f2 Mon Sep 17 00:00:00 2001 From: iTUTOR Date: Fri, 9 Jan 2026 00:07:24 +0700 Subject: [PATCH] Use the Pickle.dump directly after the dump_pickle removal in isaaclab 2.3.0 --- scripts/rsl_rl/train.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/scripts/rsl_rl/train.py b/scripts/rsl_rl/train.py index 042945e..901abbf 100644 --- a/scripts/rsl_rl/train.py +++ b/scripts/rsl_rl/train.py @@ -88,7 +88,8 @@ multi_agent_to_single_agent, ) from isaaclab.utils.dict import print_dict -from isaaclab.utils.io import dump_pickle, dump_yaml +import pickle +from isaaclab.utils.io import dump_yaml from isaaclab_rl.rsl_rl import RslRlBaseRunnerCfg, RslRlVecEnvWrapper @@ -194,8 +195,10 @@ def main(env_cfg: ManagerBasedRLEnvCfg | DirectRLEnvCfg | DirectMARLEnvCfg, agen # dump the configuration into log-directory dump_yaml(os.path.join(log_dir, "params", "env.yaml"), env_cfg) dump_yaml(os.path.join(log_dir, "params", "agent.yaml"), agent_cfg) - dump_pickle(os.path.join(log_dir, "params", "env.pkl"), env_cfg) - dump_pickle(os.path.join(log_dir, "params", "agent.pkl"), agent_cfg) + with open(os.path.join(log_dir, "params", "env.pkl"), "wb") as f: + pickle.dump(env_cfg, f) + with open(os.path.join(log_dir, "params", "agent.pkl"), "wb") as f: + pickle.dump(agent_cfg, f) # run training runner.learn(num_learning_iterations=agent_cfg.max_iterations, init_at_random_ep_len=True)