8000 fix: add MultiHeadAttention's __call__ to the stubtest allowlist · python/typeshed@a05968f · GitHub
[go: up one dir, main page]
Skip to content

Commit a05968f

Browse files
hoel-bagardhoel-bagard
committed
fix: add MultiHeadAttention's __call__ to the stubtest allowlist
1 parent 6e26f9b commit a05968f

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

‎stubs/tensorflow/@tests/stubtest_allowlist.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ tensorflow.keras.layers.*.__init__
6060
tensorflow.keras.layers.*.call
6161
tensorflow.keras.regularizers.Regularizer.__call__
6262
tensorflow.keras.constraints.Constraint.__call__
63+
tensorflow.keras.layers.MultiHeadAttention.__call__
6364

6465
# Layer class does good deal of __new__ magic and actually returns one of two different internal
6566
# types depending on tensorflow execution mode. This feels like implementation internal.

‎stubs/tensorflow/tensorflow/keras/layers/__init__.pyi

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ from typing_extensions import Self, TypeAlias
55

66
import tensorflow as tf
77
from tensorflow import Tensor, Variable, VariableAggregation, VariableSynchronization
8-
from tensorflow._aliases import AnyArray, TensorLike, TensorCompatible, DTypeLike
8+
from tensorflow._aliases import AnyArray, DTypeLike, TensorCompatible, TensorLike
99
from tensorflow.keras.activations import _Activation
1010
from tensorflow.keras.constraints import Constraint
1111
from tensorflow.keras.initializers import _Initializer

0 commit comments

Comments
 (0)
0