added check in evaluate_demo_correspondence that skips the task_progress=4

Author: adit98

evaluate_demo_correspondence.py

@@ -120,7 +120,8 @@
         raise ValueError("Must provide at least one trained model")
 
     # iterate through action_dict and visualize example signal on imitation heightmaps
-    action_keys = sorted(example_demos[0].action_dict.keys())
+    # skip last key because there is no grasp/place action associated with it
+    action_keys = sorted(example_demos[0].action_dict.keys())[:-1]
     example_actions_dict = {}
     for k in action_keys:
         if k not in example_actions_dict: