Fix grasp task

StanfordVL · Nov 15, 2023 · b1aa636 · b1aa636
1 parent f24d8ee
commit b1aa636
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/omnigibson/tasks/__init__.py b/omnigibson/tasks/__init__.py
@@ -3,3 +3,4 @@
 from omnigibson.tasks.point_navigation_task import PointNavigationTask
 from omnigibson.tasks.point_reaching_task import PointReachingTask
 from omnigibson.tasks.behavior_task import BehaviorTask
+from omnigibson.tasks.grasp_task import GraspTask
diff --git a/omnigibson/tasks/grasp_task.py b/omnigibson/tasks/grasp_task.py
@@ -1,7 +1,7 @@
 import random
 import numpy as np
 import omnigibson as og
-from omnigibson.action_primitives.starter_semantic_action_primitives import UndoableContext
+from omnigibson.action_primitives.starter_semantic_action_primitives import PlanningContext
 from omnigibson.reward_functions.grasp_reward import GraspReward
 
 from omnigibson.tasks.task_base import BaseTask