add env.py

env.py

@@ -0,0 +1,24 @@
+import os
+
+
+def make_env(n_envs: int = 1, use_async_envs: bool = False):
+    """
+    Create vectorized environments for leisaac task.
+    Args:
+        n_envs: Number of parallel environments
+        use_async_envs: Whether to use AsyncVectorEnv or SyncVectorEnv
+            (not use now)
+    Returns:
+        ManagerBasedRLEnv or DirectRLEnv inherit from gym.Env implemented in IsaacLab
+    """
+
+    # use environment variable to set config path
+    config_path = os.environ.get('LW_ENVHUB_CONFIG_PATH')
+    if config_path is None:
+        raise ValueError(
+            "Please set LW_ENVHUB_CONFIG_PATH in the environment variables")
+    print("[WARNING] ignore n_envs and use_async_envs in parameter, we use the config path to create the environment")
+
+    from lw_benchhub.utils.envhub_utils import export_env
+
+    return export_env(config_path)