summaryrefslogtreecommitdiff
path: root/utils
diff options
context:
space:
mode:
Diffstat (limited to 'utils')
-rw-r--r--utils/configuration.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/utils/configuration.py b/utils/configuration.py
index 71584c0..d7ebc5e 100644
--- a/utils/configuration.py
+++ b/utils/configuration.py
@@ -64,6 +64,8 @@ class ModelConfiguration(TypedDict):
name: str
restore_iter: int
total_iter: int
+ restore_iters: tuple[int, ...]
+ total_iters: tuple[int, ...]
class Configuration(TypedDict):