From 002b14364d1d787c65f9a9b5e3a083c87579fa09 Mon Sep 17 00:00:00 2001
From: Hoel Bagard <hoel.bagard@gmail.com>
Date: Sat, 27 Jan 2024 22:52:01 +0900
Subject: [PATCH] fix CI errors

---
 stubs/tensorflow/tensorflow/train/__init__.pyi | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/stubs/tensorflow/tensorflow/train/__init__.pyi b/stubs/tensorflow/tensorflow/train/__init__.pyi
index 547b6e9b8d07..f7ca800cd474 100644
--- 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: ...