fix imports

kscalelabs · Dec 30, 2024 · 7a0a1c3 · 7a0a1c3
1 parent 31dd379
commit 7a0a1c3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/sim/envs/base/legged_robot_config.py b/sim/envs/base/legged_robot_config.py
@@ -35,10 +35,8 @@
 
 from sim.envs.base.base_config import BaseConfig
 
-kinfer_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "third_party", "kinfer"))
+kinfer_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..", "..", "third_party", "kinfer"))
 sys.path.append(kinfer_path)
-from kinfer import proto
-
 
 class LeggedRobotCfg(BaseConfig):
     class env:
@@ -50,8 +48,6 @@ class env:
         send_timeouts = True  # send time out information to the algorithm
         episode_length_s = 20  # episode length in seconds
 
-        P = proto
-
     class terrain:
         mesh_type = "trimesh"  # "heightfield" # none, plane, heightfield or trimesh
         horizontal_scale = 0.1  # [m]

diff --git a/sim/envs/humanoids/gpr_config.py b/sim/envs/humanoids/gpr_config.py
@@ -26,7 +26,7 @@ class env(LeggedRobotCfg.env):
         episode_length_s = 24  # episode length in seconds
         use_ref_actions = False
 
-        P = LeggedRobotCfg.env.P
+        from kinfer import proto as P
 
         input_schema = P.IOSchema(
             values=[