From b6ae7c9ad3628521fbb8aaf97e45de0ac69716eb Mon Sep 17 00:00:00 2001
From: Hoel Bagard <hoel.bagard@gmail.com>
Date: Wed, 31 Jan 2024 21:52:35 +0900
Subject: [PATCH] add tf.distribute.experimental.coordinator.RemoteValue

---
 .../tensorflow/distribute/experimental/coordinator.pyi   | 9 +++++++++
 1 file changed, 9 insertions(+)
 create mode 100644 stubs/tensorflow/tensorflow/distribute/experimental/coordinator.pyi

diff --git a/stubs/tensorflow/tensorflow/distribute/experimental/coordinator.pyi b/stubs/tensorflow/tensorflow/distribute/experimental/coordinator.pyi
new file mode 100644
index 000000000000..8c8ca3139669
--- /dev/null
+++ b/stubs/tensorflow/tensorflow/distribute/experimental/coordinator.pyi
@@ -0,0 +1,9 @@
+from typing import Generic, TypeVar
+
+from tensorflow._aliases import AnyArray
+
+_Value = TypeVar("_Value", covariant=True)
+
+class RemoteValue(Generic[_Value]):
+    def fetch(self) -> AnyArray: ...
+    def get(self) -> _Value: ...