Spaces:
Runtime error
Runtime error
Commit
·
5046e86
1
Parent(s):
a2fa930
more informed err msg when obj not found
Browse files
sim.py
CHANGED
|
@@ -523,10 +523,9 @@ class PickPlaceEnv():
|
|
| 523 |
else:
|
| 524 |
obj_name = obj_name.replace('circle', 'bowl').replace('square', 'block').replace('small', '').strip()
|
| 525 |
obj_id = self.obj_name_to_id[obj_name]
|
|
|
|
| 526 |
except:
|
| 527 |
-
|
| 528 |
-
print(f'available_objects_and_id="{self.obj_name_to_id}')
|
| 529 |
-
return obj_id
|
| 530 |
|
| 531 |
def get_obj_pos(self, obj_name):
|
| 532 |
obj_name = obj_name.replace('the', '').replace('_', ' ').strip()
|
|
|
|
| 523 |
else:
|
| 524 |
obj_name = obj_name.replace('circle', 'bowl').replace('square', 'block').replace('small', '').strip()
|
| 525 |
obj_id = self.obj_name_to_id[obj_name]
|
| 526 |
+
return obj_id
|
| 527 |
except:
|
| 528 |
+
raise Exception('Object name "{}" not found'.format(obj_name))
|
|
|
|
|
|
|
| 529 |
|
| 530 |
def get_obj_pos(self, obj_name):
|
| 531 |
obj_name = obj_name.replace('the', '').replace('_', ' ').strip()
|