diff --git a/scripts/reset.py b/scripts/reset.py deleted file mode 100644 index 7c17454..0000000 --- a/scripts/reset.py +++ /dev/null @@ -1,12 +0,0 @@ -import rospy - -from active_grasp.srv import * - -rospy.init_node("reset") - -seed = rospy.ServiceProxy("seed", Seed) -reset = rospy.ServiceProxy("reset", Reset) - -seed(SeedRequest(1)) -reset(ResetRequest()) -rospy.sleep(1.0) diff --git a/test/test_sim_scene.py b/test/test_sim_scene.py new file mode 100644 index 0000000..cd4aea6 --- /dev/null +++ b/test/test_sim_scene.py @@ -0,0 +1,13 @@ +from active_grasp.simulation import Simulation + + +def main(): + gui = True + scene_id = "random" + vgn_path = "../vgn/assets/models/vgn_conv.pth" + sim = Simulation(gui, scene_id, vgn_path) + bbox = sim.reset() + + +if __name__ == "__main__": + main()