Adding type hints to call() functions in this file (#21548)

* Adding type hints to call() functions in this file * make fixup * Update src/transformers/models/marian/modeling_tf_marian.py * Update src/transformers/models/marian/modeling_tf_marian.py * Update src/transformers/models/marian/modeling_tf_marian.py * Update src/transformers/models/marian/modeling_tf_marian.py * Update src/transformers/models/marian/modeling_tf_marian.py * Update src/transformers/models/marian/modeling_tf_marian.py * Update src/transformers/models/marian/modeling_tf_marian.py * Update src/transformers/models/marian/modeling_tf_marian.py --------- Co-authored-by: Matt <rocketknight1@gmail.com> Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>

Adding type hints to call() functions in this file (#21548)
* Adding type hints to call() functions in this file * make fixup * Update src/transformers/models/marian/modeling_tf_marian.py * Update src/transformers/models/marian/modeling_tf_marian.py * Update src/transformers/models/marian/modeling_tf_marian.py * Update src/transformers/models/marian/modeling_tf_marian.py * Update src/transformers/models/marian/modeling_tf_marian.py * Update src/transformers/models/marian/modeling_tf_marian.py * Update src/transformers/models/marian/modeling_tf_marian.py * Update src/transformers/models/marian/modeling_tf_marian.py --------- Co-authored-by: Matt <rocketknight1@gmail.com> Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
bb5a2f2f · mollerup23 · GitHub · 78a53d59 · bb5a2f2f
Unverified Commit bb5a2f2f authored Feb 21, 2023 by mollerup23 Committed by GitHub Feb 21, 2023
Show whitespace changes
Inline Side-by-side

Showing with 72 additions and 72 deletions

src/transformers/models/marian/modeling_tf_marian.py src/transformers/models/marian/modeling_tf_marian.py +72 -72

No files found.
--- a/src/transformers/models/marian/modeling_tf_marian.py
+++ b/src/transformers/models/marian/modeling_tf_marian.py
@@ -708,14 +708,14 @@ class TFMarianEncoder(tf.keras.layers.Layer):
    @unpack_inputs
    def call(
        self,
-        input_ids=None,
+        input_ids: Optional[tf.Tensor] = None,
-        inputs_embeds=None,
+        inputs_embeds: Optional[tf.Tensor] = None,
-        attention_mask=None,
+        attention_mask: Optional[tf.Tensor] = None,
-        head_mask=None,
+        head_mask: Optional[tf.Tensor] = None,
-        output_attentions=None,
+        output_attentions: Optional[bool] = None,
-        output_hidden_states=None,
+        output_hidden_states: Optional[bool] = None,
-        return_dict=None,
+        return_dict: Optional[bool] = None,
-        training=False,
+        training: bool = False,
    ):
        """
        Args:
@@ -879,20 +879,20 @@ class TFMarianDecoder(tf.keras.layers.Layer):
    @unpack_inputs
    def call(
        self,
-        input_ids=None,
+        input_ids: Optional[tf.Tensor] = None,
-        inputs_embeds=None,
+        inputs_embeds: Optional[tf.Tensor] = None,
-        attention_mask=None,
+        attention_mask: Optional[tf.Tensor] = None,
-        position_ids=None,
+        position_ids: Optional[tf.Tensor] = None,
-        encoder_hidden_states=None,
+        encoder_hidden_states: Optional[tf.Tensor] = None,
-        encoder_attention_mask=None,
+        encoder_attention_mask: Optional[tf.Tensor] = None,
-        head_mask=None,
+        head_mask: Optional[tf.Tensor] = None,
-        cross_attn_head_mask=None,
+        cross_attn_head_mask: Optional[tf.Tensor] = None,
-        past_key_values=None,
+        past_key_values: Optional[Tuple[Tuple[tf.Tensor]]] = None,
-        use_cache=None,
+        use_cache: Optional[bool] = None,
-        output_attentions=None,
+        output_attentions: Optional[bool] = None,
-        output_hidden_states=None,
+        output_hidden_states: Optional[bool] = None,
-        return_dict=None,
+        return_dict: Optional[bool] = None,
-        training=False,
+        training: bool = False,
    ):
        r"""
        Args:
@@ -1115,23 +1115,23 @@ class TFMarianMainLayer(tf.keras.layers.Layer):
    @unpack_inputs
    def call(
        self,
-        input_ids=None,
+        input_ids: Optional[tf.Tensor] = None,
-        attention_mask=None,
+        attention_mask: Optional[tf.Tensor] = None,
-        decoder_input_ids=None,
+        decoder_input_ids: Optional[tf.Tensor] = None,
-        decoder_attention_mask=None,
+        decoder_attention_mask: Optional[tf.Tensor] = None,
-        decoder_position_ids=None,
+        decoder_position_ids: Optional[tf.Tensor] = None,
-        head_mask=None,
+        head_mask: Optional[tf.Tensor] = None,
-        decoder_head_mask=None,
+        decoder_head_mask: Optional[tf.Tensor] = None,
-        cross_attn_head_mask=None,
+        cross_attn_head_mask: Optional[tf.Tensor] = None,
        encoder_outputs: Optional[Union[Tuple, TFBaseModelOutput]] = None,
-        past_key_values=None,
+        past_key_values: Tuple[Tuple[tf.Tensor]] = None,
-        inputs_embeds=None,
+        inputs_embeds: Optional[tf.Tensor] = None,
-        decoder_inputs_embeds=None,
+        decoder_inputs_embeds: Optional[tf.Tensor] = None,
-        use_cache=None,
+        use_cache: Optional[bool] = None,
-        output_attentions=None,
+        output_attentions: Optional[bool] = None,
-        output_hidden_states=None,
+        output_hidden_states: Optional[bool] = None,
-        return_dict=None,
+        return_dict: Optional[bool] = None,
-        training=False,
+        training: bool = False,
        **kwargs,
    ):
        if decoder_input_ids is None and decoder_inputs_embeds is None:
@@ -1220,23 +1220,23 @@ class TFMarianModel(TFMarianPreTrainedModel):
    )
    def call(
        self,
-        input_ids=None,
+        input_ids: Optional[tf.Tensor] = None,
-        attention_mask=None,
+        attention_mask: Optional[tf.Tensor] = None,
-        decoder_input_ids=None,
+        decoder_input_ids: Optional[tf.Tensor] = None,
-        decoder_attention_mask=None,
+        decoder_attention_mask: Optional[tf.Tensor] = None,
-        decoder_position_ids=None,
+        decoder_position_ids: Optional[tf.Tensor] = None,
-        head_mask=None,
+        head_mask: Optional[tf.Tensor] = None,
-        decoder_head_mask=None,
+        decoder_head_mask: Optional[tf.Tensor] = None,
-        cross_attn_head_mask=None,
+        cross_attn_head_mask: Optional[tf.Tensor] = None,
-        encoder_outputs: Optional[Union[Tuple, TFBaseModelOutput]] = None,
+        encoder_outputs: Optional[tf.Tensor] = None,
-        past_key_values=None,
+        past_key_values: Optional[Tuple[Tuple[tf.Tensor]]] = None,
-        inputs_embeds=None,
+        inputs_embeds: Optional[tf.Tensor] = None,
-        decoder_inputs_embeds=None,
+        decoder_inputs_embeds: Optional[tf.Tensor] = None,
-        use_cache=None,
+        use_cache: Optional[bool] = None,
-        output_attentions=None,
+        output_attentions: Optional[bool] = None,
-        output_hidden_states=None,
+        output_hidden_states: Optional[bool] = None,
-        return_dict=None,
+        return_dict: Optional[bool] = None,
-        training=False,
+        training: bool = False,
        **kwargs,
    ):
        outputs = self.model(
@@ -1348,24 +1348,24 @@ class TFMarianMTModel(TFMarianPreTrainedModel, TFCausalLanguageModelingLoss):
    @add_end_docstrings(MARIAN_GENERATION_EXAMPLE)
    def call(
        self,
-        input_ids=None,
+        input_ids: Optional[tf.Tensor] = None,
-        attention_mask=None,
+        attention_mask: Optional[tf.Tensor] = None,
-        decoder_input_ids=None,
+        decoder_input_ids: Optional[tf.Tensor] = None,
-        decoder_attention_mask=None,
+        decoder_attention_mask: Optional[tf.Tensor] = None,
-        decoder_position_ids=None,
+        decoder_position_ids: Optional[tf.Tensor] = None,
-        head_mask=None,
+        head_mask: Optional[tf.Tensor] = None,
-        decoder_head_mask=None,
+        decoder_head_mask: Optional[tf.Tensor] = None,
-        cross_attn_head_mask=None,
+        cross_attn_head_mask: Optional[tf.Tensor] = None,
        encoder_outputs: Optional[TFBaseModelOutput] = None,
-        past_key_values=None,
+        past_key_values: Optional[Tuple[Tuple[tf.Tensor]]] = None,
-        inputs_embeds=None,
+        inputs_embeds: Optional[tf.Tensor] = None,
-        decoder_inputs_embeds=None,
+        decoder_inputs_embeds: Optional[tf.Tensor] = None,
-        use_cache=None,
+        use_cache: Optional[bool] = None,
-        output_attentions=None,
+        output_attentions: Optional[bool] = None,
-        output_hidden_states=None,
+        output_hidden_states: Optional[bool] = None,
-        return_dict=None,
+        return_dict: Optional[bool] = None,
-        labels=None,
+        labels: Optional[tf.Tensor] = None,
-        training=False,
+        training: bool = False,
    ):
        r"""
        labels (`tf.tensor` of shape `(batch_size, sequence_length)`, *optional*):