importing path fixed

nikhilbarhate99 · Sep 4, 2021 · c55d2b2 · c55d2b2
1 parent d9267d0
commit c55d2b2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/asset/continuous_mountain_car.py b/asset/continuous_mountain_car.py
@@ -152,7 +152,7 @@ def render_goal(self, goal, end_goal, mode='human'):
         carheight=20
 
         if self.viewer is None:
-            from gym.envs.classic_control import rendering
+            from asset import rendering
             self.viewer = rendering.Viewer(screen_width, screen_height)
             xs = np.linspace(self.min_position, self.max_position, 100)
             ys = self._height(xs)
@@ -235,7 +235,7 @@ def render_goal_2(self, goal1, goal2, end_goal, mode='human'):
 
 
         if self.viewer is None:
-            from gym.envs.classic_control import rendering
+            from asset import rendering
             self.viewer = rendering.Viewer(screen_width, screen_height)
             xs = np.linspace(self.min_position, self.max_position, 100)
             ys = self._height(xs)

diff --git a/asset/pendulum.py b/asset/pendulum.py
@@ -89,7 +89,7 @@ def render(self, mode='human'):
     def render_goal(self, goal, end_goal, mode='human'):
 
         if self.viewer is None:
-            from gym.envs.classic_control import rendering
+            from asset import rendering
             self.viewer = rendering.Viewer(500,500)
             self.viewer.set_bounds(-2.2,2.2,-2.2,2.2)
 
@@ -138,7 +138,7 @@ def render_goal(self, goal, end_goal, mode='human'):
     def render_goal_2(self, goal1, goal2, end_goal, mode='human'):
 
         if self.viewer is None:
-            from gym.envs.classic_control import rendering
+            from asset import rendering
             self.viewer = rendering.Viewer(500,500)
             self.viewer.set_bounds(-2.2,2.2,-2.2,2.2)