Skip to content

Commit 69354d7

Browse files
authored
tensorflow: add tf.distribute.experimental.coordinator.RemoteValue (#11349)
1 parent d5ff34e commit 69354d7

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

stubs/tensorflow/@tests/stubtest_allowlist.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,8 @@ tensorflow._aliases
4747
# but the real module file is completely different name (even package) and dynamically handled.
4848
# tf.initializers at runtime is <module 'keras.api._v2.keras.initializers' from '...'>
4949
tensorflow.initializers
50+
# Another cursed import magic similar to the one above.
51+
tensorflow.distribute.experimental.coordinator
5052

5153
# Layer constructor's always have **kwargs, but only allow a few specific values. PEP 692
5254
# would allow us to specify this with **kwargs and remove the need for these exceptions.
Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
from _typeshed import Incomplete
2+
from typing import Generic, TypeVar
3+
4+
from tensorflow._aliases import AnyArray
5+
6+
_Value = TypeVar("_Value", covariant=True)
7+
8+
class RemoteValue(Generic[_Value]):
9+
def fetch(self) -> AnyArray: ...
10+
def get(self) -> _Value: ...
11+
12+
def __getattr__(name: str) -> Incomplete: ...

0 commit comments

Comments
 (0)