improve wandb_checkpoint

Author: ooctipus

scripts/reinforcement_learning/rsl_rl/cli_args.py

@@ -122,9 +122,8 @@ def update_rsl_rl_cfg(agent_cfg: RslRlBaseRunnerCfg, args_cli: argparse.Namespac
                 wandb_username=args_cli.wandb_username,
             )
             agent_cfg.experiment_name = os.path.abspath(os.path.dirname(os.path.dirname(checkpoint_folder)))
-            agent_cfg.run_name = os.path.basename(os.path.dirname(checkpoint_folder))
+            agent_cfg.run_name += os.path.basename(os.path.dirname(checkpoint_folder)) + args_cli.wandb_run_id
             agent_cfg.load_checkpoint = os.path.basename(checkpoint_folder)
-            agent_cfg.run_name = args_cli.wandb_run_id
             agent_cfg.load_run = os.path.basename(os.path.dirname(checkpoint_folder))
             print(f"[INFO] Loading run from Weights & Biases: {agent_cfg.load_run}")
     return agent_cfg

source/isaaclab/isaaclab/utils/wandb_checkpoint.py

@@ -54,7 +54,7 @@ def get_model_checkpoint(
         model = models[-1]
     else:
         for remote_model in models:
-            if int(remote_model.name.split("_")[-1].split(".")[0]) == checkpoint:
+            if remote_model.name.split("_")[-1].split(".")[0] == checkpoint:
                 model = remote_model
                 break
     if model is None: