Add policy and dataset tags to WandB logging (#144)

lerobot/common/logger.py

@@ -17,9 +17,10 @@ def log_output_dir(out_dir):
 
 
 def cfg_to_group(cfg, return_list=False):
-    """Return a wandb-safe group name for logging. Optionally returns group name as list."""
+    """Return a group name for logging. Optionally returns group name as list."""
-    # lst = [cfg.task, cfg.modality, re.sub("[^0-9a-zA-Z]+", "-", cfg.exp_name)]
     lst = [
+        f"policy:{cfg.policy.name}",
+        f"dataset:{cfg.dataset_repo_id}",
         f"env:{cfg.env.name}",
         f"seed:{cfg.seed}",
     ]