Add PythonState

python · Jan 31, 2024 · e8541b9 · e8541b9
1 parent 080b155
commit e8541b9
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/stubs/tensorflow/tensorflow/train/experimental.pyi b/stubs/tensorflow/tensorflow/train/experimental.pyi
@@ -1,3 +1,13 @@
+import abc
 from _typeshed import Incomplete
+from typing import Self
+
+from tensorflow.python.trackable.base import Trackable
+
+class PythonState(Trackable, metaclass=abc.ABCMeta):
+    @abc.abstractmethod
+    def serialize(self) -> Self: ...
+    @abc.abstractmethod
+    def deserialize(self, string_value) -> str: ...
 
 def __getattr__(name: str) -> Incomplete: ...