fix CI errors

python · Jan 27, 2024 · 002b143 · 002b143
1 parent bbaa547
commit 002b143
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/stubs/tensorflow/tensorflow/train/__init__.pyi b/stubs/tensorflow/tensorflow/train/__init__.pyi
@@ -73,7 +73,7 @@ class _CheckpointLoadStatus:
     def expect_partial(self) -> Self: ...
 
 class Checkpoint:
-    def __init__(self, root: Trackable | None = None, **kwargs: Trackable): ...
+    def __init__(self, root: Trackable | None = None, **kwargs: Trackable) -> None: ...
     def write(self, file_prefix: str, options: CheckpointOptions | None = None) -> str: ...
     def read(self, file_prefix: str, options: CheckpointOptions | None = None) -> _CheckpointLoadStatus: ...
     def restore(self, file_prefix: str, options: CheckpointOptions | None = None) -> _CheckpointLoadStatus: ...
@@ -89,7 +89,7 @@ class CheckpointManager:
         step_counter: tf.Variable | None = None,
         checkpoint_interval: int | None = None,
         init_fn: Callable[[], object] | None = None,
-    ): ...
+    ) -> None: ...
     def _sweep(self) -> None: ...
 
 def latest_checkpoint(checkpoint_dir: str, latest_filename: str | None = None) -> str: ...