envs/thor_beacons.py (4 lines): - line 33: self.dist_thresh = 2e-1 # TODO: yacs - line 34: self.mask_sz = 80 # TODO: yacs - line 106: for obj in state.metadata['objects']: # don't keep all. TODO: make get obj return only visible obj list - line 112: hist['visible_objs'] = set([obj['objectId'] for obj in state.metadata['objects'] if obj['objectId'] in state.instance_masks]) # TODO: just keys rl/common/vector_env.py (1 line): - line 219: # TODO: update CALL_COMMAND for getting attribute like this kb_agent.py (1 line): - line 101: # TODO: add action names, add more actions