try removing overload

hoel-bagard · hoel-bagard · commit f9685a27c429 · 2024-01-31T23:00:11.000+09:00
diff --git a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi
@@ -277,29 +277,29 @@ class MultiHeadAttention(Layer[Any, tf.Tensor]):
         dynamic: bool = False,
         name: str | None = None,
     ) -> None: ...
-    @overload
-    def __call__(
-        self,
-        query: tf.Tensor,
-        value: tf.Tensor,
-        key: tf.Tensor | None,
-        attention_mask: tf.Tensor | None,
-        return_attention_scores: Literal[False],
-        training: bool,
-        use_causal_mask: bool,
-    ) -> tf.Tensor: ...
-    @overload
-    def __call__(
-        self,
-        query: tf.Tensor,
-        value: tf.Tensor,
-        key: tf.Tensor | None,
-        attention_mask: tf.Tensor | None,
-        return_attention_scores: Literal[True],
-        training: bool,
-        use_causal_mask: bool,
-    ) -> tuple[tf.Tensor, tf.Tensor]: ...
-    @overload
+    # @overload
+    # def __call__(
+    #     self,
+    #     query: tf.Tensor,
+    #     value: tf.Tensor,
+    #     key: tf.Tensor | None,
+    #     attention_mask: tf.Tensor | None,
+    #     return_attention_scores: Literal[False],
+    #     training: bool,
+    #     use_causal_mask: bool,
+    # ) -> tf.Tensor: ...
+    # @overload
+    # def __call__(
+    #     self,
+    #     query: tf.Tensor,
+    #     value: tf.Tensor,
+    #     key: tf.Tensor | None,
+    #     attention_mask: tf.Tensor | None,
+    #     return_attention_scores: Literal[True],
+    #     training: bool,
+    #     use_causal_mask: bool,
+    # ) -> tuple[tf.Tensor, tf.Tensor]: ...
+    # @overload
     def __call__(
         self,
         query: tf.Tensor,
