fix: use the proper entity key variable (#15)

Signed-off-by: Morgan Epp <morgane@amazon.com>
aws-deadline · Sep 1, 2023 · 8f6f9b4 · 8f6f9b4
1 parent 0d64cbd
commit 8f6f9b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/deadline_worker_agent/sessions/job_entities/job_entities.py b/src/deadline_worker_agent/sessions/job_entities/job_entities.py
@@ -267,7 +267,7 @@ def cache_entities(self, entity_identifiers: list[EntityIdentifier]):
                 # InternalServerException, ValidationException, ResourceNotFoundException,
 
                 failed_entity_key = self._entity_key(failed_entity)
-                entity_record = self._entity_record_map[entity_key]
+                entity_record = self._entity_record_map[failed_entity_key]
                 entity_record.error = failed_entity_value
                 logger.info(
                     f"BatchGetJobEntity request: {failed_entity_key} failed due to {failed_entity_value['code']}: {failed_entity_value['message']}"