From da88c9141c73e111120b0d697c0e923013bfc914 Mon Sep 17 00:00:00 2001
From: Yecheng Shao <rd_syc@unitree.com>
Date: Wed, 17 Jul 2024 17:20:50 +0800
Subject: [PATCH] add task

---
 legged_gym/scripts/play.py        |  6 +++---
 legged_gym/scripts/train.py       |  2 --
 legged_gym/utils/task_registry.py |  1 -
 setup.py                          | 11 +++++++++++
 4 files changed, 14 insertions(+), 6 deletions(-)
 create mode 100644 setup.py

diff --git a/legged_gym/scripts/play.py b/legged_gym/scripts/play.py
index 03ec2d7..fc24a88 100644
--- a/legged_gym/scripts/play.py
+++ b/legged_gym/scripts/play.py
@@ -1,7 +1,7 @@
-import os
-unitree_rl_gym_path = os.path.abspath(__file__ + "../../../../")
 import sys
-sys.path.append(unitree_rl_gym_path)
+from legged_gym import LEGGED_GYM_ROOT_DIR
+import os
+import sys
 from legged_gym import LEGGED_GYM_ROOT_DIR
 
 import isaacgym
diff --git a/legged_gym/scripts/train.py b/legged_gym/scripts/train.py
index 5d84855..aadf551 100644
--- a/legged_gym/scripts/train.py
+++ b/legged_gym/scripts/train.py
@@ -1,9 +1,7 @@
 import os
-unitree_rl_gym_path = os.path.abspath(__file__ + "../../../../")
 import numpy as np
 from datetime import datetime
 import sys
-sys.path.append(unitree_rl_gym_path)
 
 import isaacgym
 from legged_gym.envs import *
diff --git a/legged_gym/utils/task_registry.py b/legged_gym/utils/task_registry.py
index 374ffaa..5d9324f 100644
--- a/legged_gym/utils/task_registry.py
+++ b/legged_gym/utils/task_registry.py
@@ -4,7 +4,6 @@ from typing import Tuple
 import torch
 import numpy as np
 import sys
-sys.path.append("/home/unitree/unitree_rl_gym/rsl_rl")
 
 from rsl_rl.env import VecEnv
 from rsl_rl.runners import OnPolicyRunner
diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..8014c60
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,11 @@
+from setuptools import find_packages
+from distutils.core import setup
+
+setup(name='unitree_rl_gym',
+      version='1.0.0',
+      author='Unitree Robotics',
+      license="BSD-3-Clause",
+      packages=find_packages(),
+      author_email='support@unitree.com',
+      description='Template RL environments for Unitree Robots',
+      install_requires=['isaacgym', 'rsl-rl', 'matplotlib', 'numpy==1.20', 'tensorboard'])