From e351dd14b3bc3a4ce2eae4258e67a75e53a5d698 Mon Sep 17 00:00:00 2001 From: Kevin Kaspari Date: Thu, 16 Jan 2025 08:29:37 -0800 Subject: [PATCH] Add check for empty val. Merge json correctly --- tron/serialize/runstate/dynamodb_state_store.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tron/serialize/runstate/dynamodb_state_store.py b/tron/serialize/runstate/dynamodb_state_store.py index 7023b543a..35ea52da3 100644 --- a/tron/serialize/runstate/dynamodb_state_store.py +++ b/tron/serialize/runstate/dynamodb_state_store.py @@ -165,11 +165,12 @@ def _merge_items(self, first_items, remaining_items, read_json=False) -> dict: for key, item in items.items(): item.sort(key=lambda x: int(x["index"]["N"])) for val in item: - raw_items[key] += bytes(val["val"]["B"]) + if "val" in val: + raw_items[key] += bytes(val["val"]["B"]) if read_json: for json_val in item: try: - json_items[key] = json_val["json_val"]["S"] + json_items[key] += json_val["json_val"]["S"] except Exception: log.exception(f"json_val not found for key {key}") # fallback to pickled data if json_val fails to exist for any key