Fix missing attr

This commit is contained in:
Nuno Campos
2024-06-02 14:11:17 -07:00
parent 0003c74657
commit a5d413063e
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -128,7 +128,7 @@ def create_checkpoint(
channel_values=values,
channel_versions=checkpoint["channel_versions"],
versions_seen=checkpoint["versions_seen"],
pending_sends=checkpoint["pending_sends"],
pending_sends=checkpoint.get("pending_sends", []),
)
+1 -1
View File
@@ -106,7 +106,7 @@ def copy_checkpoint(checkpoint: Checkpoint) -> Checkpoint:
_seen_dict,
{k: defaultdict(int, v) for k, v in checkpoint["versions_seen"].items()},
),
pending_sends=checkpoint["pending_sends"].copy(),
pending_sends=checkpoint.get("pending_sends", []).copy(),
)