diff --git a/happypose/pose_estimators/cosypose/cosypose/lib3d/rigid_mesh_database.py b/happypose/pose_estimators/cosypose/cosypose/lib3d/rigid_mesh_database.py index e2c39747..68881f1e 100644 --- a/happypose/pose_estimators/cosypose/cosypose/lib3d/rigid_mesh_database.py +++ b/happypose/pose_estimators/cosypose/cosypose/lib3d/rigid_mesh_database.py @@ -95,8 +95,8 @@ def select(self, labels): try: ids = [self.label_to_id[label] for label in labels] except KeyError as e: - print(e) print("self.label_to_id.keys(): ", list(self.label_to_id.keys())) + raise e return Meshes( infos=[self.infos[label] for label in labels], labels=self.labels[ids],