Update env.py

env.py

@@ -208,4 +208,5 @@ def make_env(n_envs: int = 1, use_async_envs: bool = False, cfg=None) -> dict[st
         out[task][0] = env_cls(fns)
 
     # {suite_name: {task_id: VectorEnv}} 형태로 반환
-    return {"robocasa": dict(out)}
+    #return {"robocasa": dict(out)}
+    return {suite: dict(task_map) for suite, task_map in out.items()}