diff --git a/legged_gym/scripts/play.py b/legged_gym/scripts/play.py index 28aec6b..03ec2d7 100644 --- a/legged_gym/scripts/play.py +++ b/legged_gym/scripts/play.py @@ -1,7 +1,8 @@ -import sys -sys.path.append("/home/unitree/unitree_rl_gym") -from legged_gym import LEGGED_GYM_ROOT_DIR 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 isaacgym from legged_gym.envs import * diff --git a/legged_gym/scripts/train.py b/legged_gym/scripts/train.py index aa904e8..5d84855 100644 --- a/legged_gym/scripts/train.py +++ b/legged_gym/scripts/train.py @@ -1,8 +1,9 @@ -import numpy as np import os +unitree_rl_gym_path = os.path.abspath(__file__ + "../../../../") +import numpy as np from datetime import datetime import sys -sys.path.append("/home/unitree/unitree_rl_gym/") +sys.path.append(unitree_rl_gym_path) import isaacgym from legged_gym.envs import *