Rename benchmark after subset_from_split() (#221)

ServiceNow · Oct 30, 2024 · aaf7daa · aaf7daa
1 parent d33edf0
commit aaf7daa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/browsergym/experiments/src/browsergym/experiments/benchmark/base.py b/browsergym/experiments/src/browsergym/experiments/benchmark/base.py
@@ -142,6 +142,7 @@ def subset_from_split(self, split: Literal["train", "valid", "test"]):
 
         # recover the target split
         sub_benchmark = self.subset_from_regexp(split_column, regexp=f"^{split}$")
+        sub_benchmark.name = f"{self.name}_{split}"
 
         # check that the split exists (non-empty task list)
         if not sub_benchmark.env_args_list: