keras-hub-nightly 0.19.0.dev202412120352__py3-none-any.whl → 0.19.0.dev202412140350__py3-none-any.whl

keras_hub/api/layers/__init__.py

@@ -63,6 +63,7 @@ from keras_hub.src.models.segformer.segformer_image_converter import (
     SegFormerImageConverter,
 )
 from keras_hub.src.models.vgg.vgg_image_converter import VGGImageConverter
+from keras_hub.src.models.vit.vit_image_converter import ViTImageConverter
 from keras_hub.src.models.whisper.whisper_audio_converter import (
     WhisperAudioConverter,
 )

keras_hub/api/models/__init__.py

@@ -200,18 +200,18 @@ from keras_hub.src.models.image_segmenter_preprocessor import (
 )
 from keras_hub.src.models.image_to_image import ImageToImage
 from keras_hub.src.models.inpaint import Inpaint
-from keras_hub.src.models.llama3.llama3_backbone import Llama3Backbone
-from keras_hub.src.models.llama3.llama3_causal_lm import Llama3CausalLM
-from keras_hub.src.models.llama3.llama3_causal_lm_preprocessor import (
-    Llama3CausalLMPreprocessor,
-)
-from keras_hub.src.models.llama3.llama3_tokenizer import Llama3Tokenizer
 from keras_hub.src.models.llama.llama_backbone import LlamaBackbone
 from keras_hub.src.models.llama.llama_causal_lm import LlamaCausalLM
 from keras_hub.src.models.llama.llama_causal_lm_preprocessor import (
     LlamaCausalLMPreprocessor,
 )
 from keras_hub.src.models.llama.llama_tokenizer import LlamaTokenizer
+from keras_hub.src.models.llama3.llama3_backbone import Llama3Backbone
+from keras_hub.src.models.llama3.llama3_causal_lm import Llama3CausalLM
+from keras_hub.src.models.llama3.llama3_causal_lm_preprocessor import (
+    Llama3CausalLMPreprocessor,
+)
+from keras_hub.src.models.llama3.llama3_tokenizer import Llama3Tokenizer
 from keras_hub.src.models.masked_lm import MaskedLM
 from keras_hub.src.models.masked_lm_preprocessor import MaskedLMPreprocessor
 from keras_hub.src.models.mistral.mistral_backbone import MistralBackbone
@@ -330,6 +330,11 @@ from keras_hub.src.models.vgg.vgg_image_classifier import VGGImageClassifier
 from keras_hub.src.models.vgg.vgg_image_classifier_preprocessor import (
     VGGImageClassifierPreprocessor,
 )
+from keras_hub.src.models.vit.vit_backbone import ViTBackbone
+from keras_hub.src.models.vit.vit_image_classifier import ViTImageClassifier
+from keras_hub.src.models.vit.vit_image_classifier_preprocessor import (
+    ViTImageClassifierPreprocessor,
+)
 from keras_hub.src.models.vit_det.vit_det_backbone import ViTDetBackbone
 from keras_hub.src.models.whisper.whisper_backbone import WhisperBackbone
 from keras_hub.src.models.whisper.whisper_tokenizer import WhisperTokenizer

keras_hub/api/tokenizers/__init__.py

@@ -21,8 +21,8 @@ from keras_hub.src.models.falcon.falcon_tokenizer import FalconTokenizer
 from keras_hub.src.models.gemma.gemma_tokenizer import GemmaTokenizer
 from keras_hub.src.models.gpt2.gpt2_tokenizer import GPT2Tokenizer
 from keras_hub.src.models.gpt_neo_x.gpt_neo_x_tokenizer import GPTNeoXTokenizer
-from keras_hub.src.models.llama3.llama3_tokenizer import Llama3Tokenizer
 from keras_hub.src.models.llama.llama_tokenizer import LlamaTokenizer
+from keras_hub.src.models.llama3.llama3_tokenizer import Llama3Tokenizer
 from keras_hub.src.models.mistral.mistral_tokenizer import MistralTokenizer
 from keras_hub.src.models.opt.opt_tokenizer import OPTTokenizer
 from keras_hub.src.models.pali_gemma.pali_gemma_tokenizer import (

keras_hub/src/bounding_box/converters.py

@@ -73,8 +73,8 @@ def encode_box_to_deltas(
 
     if encoding_format not in ["center_xywh", "center_yxhw"]:
         raise ValueError(
-            "`encoding_format` should be one of 'center_xywh' or 'center_yxhw', "
-            f"got {encoding_format}"
+            "`encoding_format` should be one of 'center_xywh' or "
+            f"'center_yxhw', got {encoding_format}"
         )
 
     encoded_anchors = convert_format(

keras_hub/src/layers/modeling/f_net_encoder.py

@@ -66,7 +66,7 @@ class FNetEncoder(keras.layers.Layer):
         layer_norm_epsilon=1e-5,
         kernel_initializer="glorot_uniform",
         bias_initializer="zeros",
-        **kwargs
+        **kwargs,
     ):
         super().__init__(**kwargs)
         self.intermediate_dim = intermediate_dim

keras_hub/src/layers/modeling/masked_lm_head.py

@@ -34,7 +34,8 @@ class MaskedLMHead(keras.layers.Layer):
         token_embedding: Optional. A `keras_hub.layers.ReversibleEmbedding`
             instance. If passed, the layer will be used to project from the
             `hidden_dim` of the model to the output `vocabulary_size`.
-        intermediate_activation: The activation function of intermediate dense layer.
+        intermediate_activation: The activation function of intermediate dense
+            layer.
         activation: The activation function for the outputs of the layer.
             Usually either `None` (return logits), or `"softmax"`
             (return probabilities).

keras_hub/src/layers/modeling/rms_normalization.py

@@ -6,10 +6,11 @@ from keras_hub.src.api_export import keras_hub_export
 
 @keras_hub_export("keras_hub.layers.RMSNormalization")
 class RMSNormalization(keras.layers.Layer):
-    """
-    Root Mean Square (RMS) Normalization layer.
+    """Root Mean Square (RMS) Normalization layer.
+
     This layer normalizes the input tensor based on its RMS value and applies
     a learned scaling factor.
+
     Args:
         input_dim: int. The dimensionality of the input tensor.
     """
@@ -21,12 +22,13 @@ class RMSNormalization(keras.layers.Layer):
         )
 
     def call(self, x):
-        """
-        Applies RMS normalization to the input tensor.
+        """Applies RMS normalization to the input tensor.
+
         Args:
-            x: KerasTensor. Input tensor of shape (batch_size, input_dim).
+            x: Input tensor of shape (batch_size, input_dim).
+
         Returns:
-            KerasTensor: The RMS-normalized tensor of the same shape (batch_size, input_dim),
+            The RMS-normalized tensor of the same shape (batch_size, input_dim),
             scaled by the learned `scale` parameter.
         """
         x = ops.cast(x, float)

keras_hub/src/layers/modeling/rotary_embedding.py

@@ -11,7 +11,8 @@ class RotaryEmbedding(keras.layers.Layer):
     This layer encodes absolute positional information with a rotation
     matrix. It calculates the rotary encoding with a mix of sine and
     cosine functions with geometrically increasing wavelengths.
-    Defined and formulated in [RoFormer: Enhanced Transformer with Rotary Position Embedding](https://arxiv.org/abs/2104.09864v4).
+    Defined and formulated in
+    [RoFormer: Enhanced Transformer with Rotary Position Embedding](https://arxiv.org/abs/2104.09864v4).
     The input must be a tensor with shape a sequence dimension and a feature
     dimension. Typically, this will either an input with shape
     `(batch_size, sequence_length, feature_length)` or
@@ -65,7 +66,7 @@ class RotaryEmbedding(keras.layers.Layer):
         scaling_factor=1.0,
         sequence_axis=1,
         feature_axis=-1,
-        **kwargs
+        **kwargs,
     ):
         super().__init__(**kwargs)
         self.max_wavelength = max_wavelength

keras_hub/src/layers/modeling/token_and_position_embedding.py

@@ -58,7 +58,7 @@ class TokenAndPositionEmbedding(keras.layers.Layer):
         tie_weights=True,
         embeddings_initializer="uniform",
         mask_zero=False,
-        **kwargs
+        **kwargs,
     ):
         super().__init__(**kwargs)
         if vocabulary_size is None:

keras_hub/src/layers/modeling/transformer_decoder.py

@@ -5,12 +5,13 @@ from keras_hub.src.api_export import keras_hub_export
 from keras_hub.src.layers.modeling.cached_multi_head_attention import (
     CachedMultiHeadAttention,
 )
-from keras_hub.src.utils.keras_utils import clone_initializer
-
-from keras_hub.src.layers.modeling.transformer_layer_utils import (  # isort:skip
+from keras_hub.src.layers.modeling.transformer_layer_utils import (
     compute_causal_mask,
+)
+from keras_hub.src.layers.modeling.transformer_layer_utils import (
     merge_padding_and_attention_mask,
 )
+from keras_hub.src.utils.keras_utils import clone_initializer
 
 
 @keras_hub_export("keras_hub.layers.TransformerDecoder")
@@ -265,13 +266,13 @@ class TransformerDecoder(keras.layers.Layer):
                 `[batch_size, decoder_sequence_length]`.
             decoder_attention_mask: a boolean Tensor. Customized decoder
                 sequence mask, must be of shape
-                `[batch_size, decoder_sequence_length, decoder_sequence_length]`.
+                `[batch_size, decoder_sequence_length, decoder_sequence_length]`
             encoder_padding_mask: a boolean Tensor, the padding mask of encoder
                 sequence, must be of shape
                 `[batch_size, encoder_sequence_length]`.
             encoder_attention_mask: a boolean Tensor. Customized encoder
                 sequence mask, must be of shape
-                `[batch_size, encoder_sequence_length, encoder_sequence_length]`.
+                `[batch_size, encoder_sequence_length, encoder_sequence_length]`
             self_attention_cache: a dense float Tensor. The cache of key/values
                 pairs in the self-attention layer. Has shape
                 `[batch_size, 2, max_seq_len, num_heads, key_dims]`.
@@ -435,7 +436,8 @@ class TransformerDecoder(keras.layers.Layer):
             input_length = output_length = ops.shape(decoder_sequence)[1]
             # We need to handle a rectangular causal mask when doing cached
             # decoding. For generative inference, `decoder_sequence` will
-            # generally be length 1, and `cache` will be the full generation length.
+            # generally be length 1, and `cache` will be the full generation
+            # length.
             if self_attention_cache is not None:
                 input_length = ops.shape(self_attention_cache)[2]
 

keras_hub/src/layers/modeling/transformer_encoder.py

@@ -190,7 +190,9 @@ class TransformerEncoder(keras.layers.Layer):
                 [batch_size, sequence_length, sequence_length].
             training: a boolean indicating whether the layer should behave in
                 training mode or in inference mode.
-            return_attention_scores: a boolean indicating whether the output should be `(attention_output, attention_scores)` if `True` or `attention_output` if `False`. Defaults to `False`.
+            return_attention_scores: a boolean indicating whether the output
+                should be `(attention_output, attention_scores)` if `True` or
+                `attention_output` if `False`. Defaults to `False`.
 
         Returns:
             A Tensor of the same shape as the `inputs`.

keras_hub/src/metrics/bleu.py

@@ -164,7 +164,7 @@ class Bleu(keras.metrics.Metric):
         return inputs
 
     def _get_ngrams(self, segment, max_order):
-        """Extracts all n-grams up to a given maximum order from an input segment.
+        """Extracts all n-grams up to a given maximum order from an input.
 
         Uses Python ops. Inspired from
         https://github.com/tensorflow/nmt/blob/master/nmt/scripts/bleu.py.

keras_hub/src/models/albert/albert_text_classifier.py

@@ -20,10 +20,10 @@ from keras_hub.src.models.text_classifier import TextClassifier
 class AlbertTextClassifier(TextClassifier):
     """An end-to-end ALBERT model for classification tasks
 
-    This model attaches a classification head to a `keras_hub.model.AlbertBackbone`
-    backbone, mapping from the backbone outputs to logit output suitable for
-    a classification task. For usage of this model with pre-trained weights, see
-    the `from_preset()` method.
+    This model attaches a classification head to a
+    `keras_hub.model.AlbertBackbone` backbone, mapping from the backbone outputs
+    to logit output suitable for a classification task. For usage of this model
+    with pre-trained weights, see the `from_preset()` method.
 
     This model can optionally be configured with a `preprocessor` layer, in
     which case it will automatically apply preprocessing to raw inputs during
@@ -36,9 +36,9 @@ class AlbertTextClassifier(TextClassifier):
     Args:
         backbone: A `keras_hub.models.AlertBackbone` instance.
         num_classes: int. Number of classes to predict.
-        preprocessor: A `keras_hub.models.AlbertTextClassifierPreprocessor` or `None`. If
-            `None`, this model will not apply preprocessing, and inputs should
-            be preprocessed before calling the model.
+        preprocessor: A `keras_hub.models.AlbertTextClassifierPreprocessor` or
+            `None`. If `None`, this model will not apply preprocessing, and
+            inputs should be preprocessed before calling the model.
         activation: Optional `str` or callable. The
             activation function to use on the model outputs. Set
             `activation="softmax"` to return output probabilities.

keras_hub/src/models/bart/bart_backbone.py

@@ -22,9 +22,9 @@ class BartBackbone(Backbone):
     described in
     ["BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehension"](https://arxiv.org/abs/1910.13461).
 
-    The default constructor gives a fully customizable, randomly initialized BART
-    model with any number of layers, heads, and embedding dimensions. To load
-    preset architectures and weights, use the `from_preset` constructor.
+    The default constructor gives a fully customizable, randomly initialized
+    BART model with any number of layers, heads, and embedding dimensions. To
+    load preset architectures and weights, use the `from_preset` constructor.
 
     Disclaimer: Pre-trained models are provided on an "as is" basis, without
     warranties or conditions of any kind. The underlying model is provided by a
@@ -78,7 +78,7 @@ class BartBackbone(Backbone):
     )
     output = model(input_data)
     ```
-    """
+    """  # noqa: E501
 
     def __init__(
         self,

keras_hub/src/models/bart/bart_seq_2_seq_lm.py

@@ -60,7 +60,8 @@ class BartSeq2SeqLM(Seq2SeqLM):
     bart_lm.generate("The quick brown fox", max_length=30)
     ```
 
-    Use `generate()` with encoder inputs and an incomplete decoder input (prompt).
+    Use `generate()` with encoder inputs and an incomplete decoder input
+    (prompt).
     ```python
     bart_lm = keras_hub.models.BartSeq2SeqLM.from_preset("bart_base_en")
     bart_lm.generate(
@@ -79,10 +80,10 @@ class BartSeq2SeqLM(Seq2SeqLM):
     prompt = {
         "encoder_token_ids": np.array([[0, 133, 2119, 6219, 23602, 2, 1, 1]]),
         "encoder_padding_mask": np.array(
-            [[True, True, True, True, True, True, False, False]]
+            [[1, 1, 1, 1, 1, 1, 0, 0]]
         ),
         "decoder_token_ids": np.array([[2, 0, 133, 1769, 2, 1, 1]]),
-        "decoder_padding_mask": np.array([[True, True, True, True, False, False]])
+        "decoder_padding_mask": np.array([[1, 1, 1, 1, 0, 0]])
     }
 
     bart_lm = keras_hub.models.BartSeq2SeqLM.from_preset(
@@ -95,7 +96,7 @@ class BartSeq2SeqLM(Seq2SeqLM):
     Call `fit()` on a single batch.
     ```python
     features = {
-        "encoder_text": ["The quick brown fox jumped.", "I forgot my homework."],
+        "encoder_text": ["The quick fox jumped.", "I forgot my homework."],
         "decoder_text": ["The fast hazel fox leapt.", "I forgot my assignment."]
     }
     bart_lm = keras_hub.models.BartSeq2SeqLM.from_preset("bart_base_en")
@@ -195,7 +196,7 @@ class BartSeq2SeqLM(Seq2SeqLM):
         cross_attention_cache=None,
         cross_attention_cache_update_index=None,
     ):
-        """Forward pass with a key/value caches for generative decoding..
+        """Forward pass with a key/value caches for generative decoding.
 
         `call_decoder_with_cache` adds an additional inference-time forward pass
         for the model for seq2seq text generation. Unlike calling the model
@@ -241,7 +242,7 @@ class BartSeq2SeqLM(Seq2SeqLM):
             key/value cache in the decoder's self-attention layer and
             `cross_attention_cache` is the key/value cache in the decoder's
             cross-attention layer.
-        """
+        """  # noqa: E501
         # Embedding layers.
         tokens = self.backbone.token_embedding(decoder_token_ids)
         positions = self.backbone.decoder_position_embedding(
@@ -331,7 +332,7 @@ class BartSeq2SeqLM(Seq2SeqLM):
     def _build_cache(
         self, encoder_token_ids, encoder_padding_mask, decoder_token_ids
     ):
-        """Builds the self-attention cache and the cross-attention cache (key/value pairs)."""
+        """Builds the self-attention cache and the cross-attention cache."""
         encoder_hidden_states = self.call_encoder(
             token_ids=encoder_token_ids, padding_mask=encoder_padding_mask
         )
@@ -417,7 +418,7 @@ class BartSeq2SeqLM(Seq2SeqLM):
             prompt = ops.slice(prompt, [0, cache_index], [num_samples, 1])
 
             def repeat_tensor(x):
-                """Repeats tensors along batch axis to match dim for beam search."""
+                """Repeats along batch axis to match dim for beam search."""
                 if ops.shape(x)[0] == num_samples:
                     return x
                 return ops.repeat(x, repeats=num_samples // batch_size, axis=0)

keras_hub/src/models/bert/bert_presets.py

@@ -69,7 +69,8 @@ backbone_presets = {
     "bert_base_multi": {
         "metadata": {
             "description": (
-                "12-layer BERT model where case is maintained. Trained on trained on Wikipedias of 104 languages"
+                "12-layer BERT model where case is maintained. Trained on "
+                "trained on Wikipedias of 104 languages"
             ),
             "params": 177853440,
             "path": "bert",
@@ -101,7 +102,8 @@ backbone_presets = {
     "bert_tiny_en_uncased_sst2": {
         "metadata": {
             "description": (
-                "The bert_tiny_en_uncased backbone model fine-tuned on the SST-2 sentiment analysis dataset."
+                "The bert_tiny_en_uncased backbone model fine-tuned on the "
+                "SST-2 sentiment analysis dataset."
             ),
             "params": 4385920,
             "path": "bert",

keras_hub/src/models/bert/bert_text_classifier.py

@@ -34,9 +34,9 @@ class BertTextClassifier(TextClassifier):
     Args:
         backbone: A `keras_hub.models.BertBackbone` instance.
         num_classes: int. Number of classes to predict.
-        preprocessor: A `keras_hub.models.BertTextClassifierPreprocessor` or `None`. If
-            `None`, this model will not apply preprocessing, and inputs should
-            be preprocessed before calling the model.
+        preprocessor: A `keras_hub.models.BertTextClassifierPreprocessor` or
+            `None`. If `None`, this model will not apply preprocessing, and
+            inputs should be preprocessed before calling the model.
         activation: Optional `str` or callable. The
             activation function to use on the model outputs. Set
             `activation="softmax"` to return output probabilities.

keras_hub/src/models/causal_lm.py

@@ -303,16 +303,18 @@ class CausalLM(Task):
                 `preprocessor`. If `preprocessor` is `None`, `inputs` should be
                 should be padded to the desired maximum length and this argument
                 will be ignored.
-            stop_token_ids: Optional. `None`, "auto", or tuple of token ids. Defaults
-                to "auto" which uses the `preprocessor.tokenizer.end_token_id`.
-                Not specifying a processor will produce an error. None stops
-                generation after generating `max_length` tokens. You may also
-                specify a list of token id's the model should stop on. Note that
-                sequences of tokens will each be interpreted as a stop token,
-                multi-token stop sequences are not supported.
-            strip_prompt: Optional. By default, generate() returns the full prompt
-                followed by its completion generated by the model. If this option
-                is set to True, only the newly generated text is returned.
+            stop_token_ids: Optional. `None`, "auto", or tuple of token ids.
+                Defaults to "auto" which uses the
+                `preprocessor.tokenizer.end_token_id`. Not specifying a
+                processor will produce an error. None stops generation after
+                generating `max_length` tokens. You may also specify a list of
+                token id's the model should stop on. Note that sequences of
+                tokens will each be interpreted as a stop token, multi-token
+                stop sequences are not supported.
+            strip_prompt: Optional. By default, generate() returns the full
+                prompt followed by its completion generated by the model. If
+                this option is set to True, only the newly generated text is
+                returned.
         """
         # Setup our three main passes.
         # 1. Optionally preprocessing strings to dense integer tensors.
@@ -322,10 +324,11 @@ class CausalLM(Task):
 
         if self.preprocessor is None and stop_token_ids == "auto":
             raise ValueError(
-                'A `preprocessor` must be attached to the model if `stop_token_ids="auto"`. '
-                "Currently `preprocessor=None`. To call `generate()` with preprocessing "
-                "detached, either pass `stop_token_ids=None` to always generate until "
-                "`max_length` or pass a tuple of token ids that should terminate generation "
+                "A `preprocessor` must be attached to the model if "
+                '`stop_token_ids="auto"`. Currently `preprocessor=None`. To '
+                "call `generate()` with preprocessing detached, either pass "
+                "`stop_token_ids=None` to always generate until `max_length` "
+                "or pass a tuple of token ids that should terminate generation "
                 "as `stop_token_ids`."
             )
         elif stop_token_ids == "auto":
@@ -365,7 +368,8 @@ class CausalLM(Task):
             y["padding_mask"] = ops.logical_xor(
                 roll_sequence(prompt_mask), roll_sequence(x["padding_mask"])
             )
-            # we assume the mask is enough and there is no need to zero-out the values
+            # we assume the mask is enough and there is no need to zero-out the
+            # values
             y["token_ids"] = roll_sequence(x["token_ids"])
 
             return y

keras_hub/src/models/clip/clip_vision_embedding.py

@@ -12,7 +12,7 @@ class CLIPVisionEmbedding(layers.Layer):
         image_size,
         data_format=None,
         dtype=None,
-        **kwargs
+        **kwargs,
     ):
         super().__init__(dtype=dtype, **kwargs)
         self.hidden_dim = int(hidden_dim)

keras_hub/src/models/csp_darknet/csp_darknet_backbone.py

@@ -360,7 +360,8 @@ def apply_cross_stage_partial(
     """
 
     if name is None:
-        name = f"cross_stage_partial_{keras.backend.get_uid('cross_stage_partial')}"
+        uid = keras.backend.get_uid("cross_stage_partial")
+        name = f"cross_stage_partial_{uid}"
 
     def apply(inputs):
         hidden_channels = filters // 2

keras_hub/src/models/deberta_v3/deberta_v3_backbone.py

@@ -83,7 +83,7 @@ class DebertaV3Backbone(Backbone):
     # Call the model on the input data.
     model(input_data)
     ```
-    """
+    """  # noqa: E501
 
     def __init__(
         self,

keras_hub/src/models/deberta_v3/deberta_v3_text_classifier.py

@@ -7,7 +7,7 @@ from keras_hub.src.models.deberta_v3.deberta_v3_backbone import (
 from keras_hub.src.models.deberta_v3.deberta_v3_backbone import (
     deberta_kernel_initializer,
 )
-from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier_preprocessor import (
+from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier_preprocessor import (  # noqa: E501
     DebertaV3TextClassifierPreprocessor,
 )
 from keras_hub.src.models.text_classifier import TextClassifier
@@ -43,9 +43,9 @@ class DebertaV3TextClassifier(TextClassifier):
     Args:
         backbone: A `keras_hub.models.DebertaV3` instance.
         num_classes: int. Number of classes to predict.
-        preprocessor: A `keras_hub.models.DebertaV3TextClassifierPreprocessor` or `None`. If
-            `None`, this model will not apply preprocessing, and inputs should
-            be preprocessed before calling the model.
+        preprocessor: A `keras_hub.models.DebertaV3TextClassifierPreprocessor`
+            or `None`. If `None`, this model will not apply preprocessing, and
+            inputs should be preprocessed before calling the model.
         activation: Optional `str` or callable. The
             activation function to use on the model outputs. Set
             `activation="softmax"` to return output probabilities.

keras_hub/src/models/deberta_v3/disentangled_attention_encoder.py

@@ -45,7 +45,7 @@ class DisentangledAttentionEncoder(keras.layers.Layer):
         bias_initializer: string or `keras.initializers` initializer.
             The bias initializer for the dense and disentangled
             self-attention layers. Defaults to `"zeros"`.
-    """
+    """  # noqa: E501
 
     def __init__(
         self,
@@ -58,7 +58,7 @@ class DisentangledAttentionEncoder(keras.layers.Layer):
         layer_norm_epsilon=1e-05,
         kernel_initializer="glorot_uniform",
         bias_initializer="zeros",
-        **kwargs
+        **kwargs,
     ):
         super().__init__(**kwargs)
         self.intermediate_dim = intermediate_dim
@@ -145,8 +145,8 @@ class DisentangledAttentionEncoder(keras.layers.Layer):
         """Forward pass of `DisentangledAttentionEncoder`.
 
         Args:
-            inputs: a Tensor. The input data to `DisentangledAttentionEncoder`, should be
-                of shape [batch_size, sequence_length, hidden_dim].
+            inputs: a Tensor. The input data to `DisentangledAttentionEncoder`,
+                should be of shape [batch_size, sequence_length, hidden_dim].
             rel_embeddings: a Tensor. The relative position embedding matrix,
                 should be of shape `[batch_size, 2 * bucket_size, hidden_dim]`.
             padding_mask: a boolean Tensor. It indicates if the token should be

keras_hub/src/models/deberta_v3/disentangled_self_attention.py

@@ -31,7 +31,7 @@ class DisentangledSelfAttention(keras.layers.Layer):
         bias_initializer: string or `keras.initializers` initializer.
             The bias initializer for the dense layers.
             Defaults to `"zeros"`.
-    """
+    """  # noqa: E501
 
     def __init__(
         self,
@@ -363,7 +363,8 @@ class DisentangledSelfAttention(keras.layers.Layer):
             training=training,
         )
 
-        # Reshape `attention_output` to `(batch_size, sequence_length, hidden_dim)`.
+        # Reshape `attention_output` to
+        # `(batch_size, sequence_length, hidden_dim)`.
         attention_output = ops.reshape(
             attention_output,
             [

keras_hub/src/models/deberta_v3/relative_embedding.py

@@ -20,7 +20,7 @@ class RelativeEmbedding(keras.layers.Layer):
         kernel_initializer: string or `keras.initializers` initializer.
             The kernel initializer for the dense embedding.
             Defaults to `"glorot_uniform"`.
-    """
+    """  # noqa: E501
 
     def __init__(
         self,

keras_hub/src/models/deeplab_v3/deeplab_v3_backbone.py

@@ -12,10 +12,10 @@ class DeepLabV3Backbone(Backbone):
     """DeepLabV3 & DeepLabV3Plus architecture for semantic segmentation.
 
     This class implements a DeepLabV3 & DeepLabV3Plus architecture as described
-    in [Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation](
-        https://arxiv.org/abs/1802.02611)(ECCV 2018)
-    and [Rethinking Atrous Convolution for Semantic Image Segmentation](
-        https://arxiv.org/abs/1706.05587)(CVPR 2017)
+    in [Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation](https://arxiv.org/abs/1802.02611)
+    (ECCV 2018)
+    and [Rethinking Atrous Convolution for Semantic Image Segmentation](https://arxiv.org/abs/1706.05587)
+    (CVPR 2017)
 
     Args:
         image_encoder: `keras.Model`. An instance that is used as a feature
@@ -35,20 +35,23 @@ class DeepLabV3Backbone(Backbone):
             layer resolution should match with the `low_level_feature`s layer
             resolution to concatenate both the layers for combined encoder
             outputs.
-        dilation_rates: list. A `list` of integers for parallel dilated conv applied to
-        `SpatialPyramidPooling`. Usually a
+        dilation_rates: list. A `list` of integers for parallel dilated conv
+            applied to `SpatialPyramidPooling`. Usually a
             sample choice of rates are `[6, 12, 18]`.
-        low_level_feature_key: str optional. A layer level to extract the feature
-            from one of the key from the `image_encoder`s `pyramid_outputs`
-            property such as  "P2", "P3" etc which will be the Decoder block.
-            Required only when the DeepLabV3Plus architecture needs to be applied.
+        low_level_feature_key: str optional. A layer level to extract the
+            feature from one of the key from the `image_encoder`s
+            `pyramid_outputs` property such as  "P2", "P3" etc which will be the
+            Decoder block. Required only when the DeepLabV3Plus architecture
+            needs to be applied.
         image_shape: tuple. The input shape without the batch size.
             Defaults to `(None, None, 3)`.
 
     Example:
     ```python
     # Load a trained backbone to extract features from it's `pyramid_outputs`.
-    image_encoder = keras_hub.models.ResNetBackbone.from_preset("resnet_50_imagenet")
+    image_encoder = keras_hub.models.ResNetBackbone.from_preset(
+        "resnet_50_imagenet"
+    )
 
     model = keras_hub.models.DeepLabV3Backbone(
         image_encoder=image_encoder,
@@ -59,7 +62,7 @@ class DeepLabV3Backbone(Backbone):
         dilation_rates = [6, 12, 18]
     )
     ```
-    """
+    """  # noqa: E501
 
     def __init__(
         self,
@@ -74,7 +77,8 @@ class DeepLabV3Backbone(Backbone):
     ):
         if not isinstance(image_encoder, keras.Model):
             raise ValueError(
-                "Argument `image_encoder` must be a `keras.Model` instance. Received instead "
+                "Argument `image_encoder` must be a `keras.Model` instance. "
+                "Received instead "
                 f"{image_encoder} (of type {type(image_encoder)})."
             )
         data_format = keras.config.image_data_format()

keras_hub/src/models/deeplab_v3/deeplab_v3_presets.py

@@ -4,9 +4,10 @@ backbone_presets = {
     "deeplab_v3_plus_resnet50_pascalvoc": {
         "metadata": {
             "description": (
-                "DeepLabV3+ model with ResNet50 as image encoder and trained on "
-                "augmented Pascal VOC dataset by Semantic Boundaries Dataset(SBD)"
-                "which is having categorical accuracy of 90.01 and 0.63 Mean IoU."
+                "DeepLabV3+ model with ResNet50 as image encoder and trained "
+                "on augmented Pascal VOC dataset by Semantic Boundaries "
+                "Dataset(SBD) which is having categorical accuracy of 90.01 "
+                "and 0.63 Mean IoU."
             ),
             "params": 39190656,
             "path": "deeplab_v3",

keras_hub/src/models/deeplab_v3/deeplab_v3_segmenter.py

@@ -4,7 +4,7 @@ from keras_hub.src.api_export import keras_hub_export
 from keras_hub.src.models.deeplab_v3.deeplab_v3_backbone import (
     DeepLabV3Backbone,
 )
-from keras_hub.src.models.deeplab_v3.deeplab_v3_image_segmeter_preprocessor import (
+from keras_hub.src.models.deeplab_v3.deeplab_v3_image_segmeter_preprocessor import (  # noqa: E501
     DeepLabV3ImageSegmenterPreprocessor,
 )
 from keras_hub.src.models.image_segmenter import ImageSegmenter