Small touch to the visualization

2021-08-30 12:51:38 +02:00
parent 9f0e2789cf
commit 98abe3789d
3 changed files with 10 additions and 3 deletions
--- a/active_grasp/policy.py
+++ b/active_grasp/policy.py
@@ -116,6 +116,8 @@ class MultiViewPolicy(Policy):

        if len(grasps) > 0:
            self.best_grasp = grasps[0]
+        else:
+            self.best_grasp = None

        self.vis.grasps(self.base_frame, grasps, scores)