Add missing branch

This commit is contained in:
Nuno Campos
2024-09-18 13:18:37 -07:00
parent b062b2462f
commit efb572f15b
2 changed files with 4 additions and 0 deletions
@@ -510,6 +510,8 @@ class SqliteSaver(BaseCheckpointSaver):
"""
if current is None:
current_v = 0
elif isinstance(current, int):
current_v = current
else:
current_v = int(current.split(".")[0])
next_v = current_v + 1
@@ -515,6 +515,8 @@ class AsyncSqliteSaver(BaseCheckpointSaver):
"""
if current is None:
current_v = 0
elif isinstance(current, int):
current_v = current
else:
current_v = int(current.split(".")[0])
next_v = current_v + 1