Update env.py

env.py

@@ -224,11 +224,18 @@ def make_env(n_envs: int = 1, use_async_envs: bool = False, cfg=None) -> dict[st
     combined_tasks = {**TARGET_TASKS, **PRETRAINING_TASKS}
 
     # 벤치마크인지 단일 태스크인지 구분
-    if task_name in combined_tasks:
-        task_names = combined_tasks[task_name]
-        gym_kwargs["split"] = "target" if task_name in TARGET_TASKS else "pretrain"
+    parts = [t.strip() for t in task_name.split(",")]
+    if len(parts) == 1 and parts[0] in combined_tasks:
+        task_names = combined_tasks[parts[0]]
+        if gym_kwargs.get("split") is None:
+            gym_kwargs["split"] = "target" if parts[0] in TARGET_TASKS else "pretrain"
     else:
-        task_names = [t.strip() for t in task_name.split(",")]
+        task_names = []
+        for part in parts:
+            if part in combined_tasks:
+                task_names.extend(combined_tasks[part])
+            else:
+                task_names.append(part)
         
 
     out = defaultdict(dict)