keras-hub-nightly 0.19.0.dev202501080345__tar.gz → 0.19.0.dev202501150344__tar.gz

{keras_hub_nightly-0.19.0.dev202501080345 → keras_hub_nightly-0.19.0.dev202501150344}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: keras-hub-nightly
-Version: 0.19.0.dev202501080345
+Version: 0.19.0.dev202501150344
 Summary: Industry-strength Natural Language Processing extensions for Keras.
 Home-page: https://github.com/keras-team/keras-hub
 Author: Keras team
@@ -31,6 +31,17 @@ Requires-Dist: tensorflow-text
 Provides-Extra: extras
 Requires-Dist: rouge-score; extra == "extras"
 Requires-Dist: sentencepiece; extra == "extras"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 # KerasHub: Multi-framework Pretrained Models
 [![](https://github.com/keras-team/keras-hub/workflows/Tests/badge.svg?branch=master)](https://github.com/keras-team/keras-hub/actions?query=workflow%3ATests+branch%3Amaster)

{keras_hub_nightly-0.19.0.dev202501080345 → keras_hub_nightly-0.19.0.dev202501150344}/keras_hub/src/metrics/bleu.py

@@ -329,8 +329,9 @@ class Bleu(keras.metrics.Metric):
                 return tf.squeeze(inputs, axis=-1)
             else:
                 raise ValueError(
-                    f"{tensor_name} must be of rank {base_rank}, {base_rank+1} "
-                    f"or {base_rank+2}. Found rank: {inputs.shape.rank}"
+                    f"{tensor_name} must be of rank {base_rank}, "
+                    f"{base_rank + 1}, or {base_rank + 2}. "
+                    f"Found rank: {inputs.shape.rank}"
                 )
 
         y_true = validate_and_fix_rank(y_true, "y_true", 1)

{keras_hub_nightly-0.19.0.dev202501080345 → keras_hub_nightly-0.19.0.dev202501150344}/keras_hub/src/models/basnet/basnet_backbone.py

@@ -219,7 +219,7 @@ def get_resnet_block(_resnet, block_num):
     else:
         x = _resnet.pyramid_outputs[extractor_levels[block_num - 1]]
     y = _resnet.get_layer(
-        f"stack{block_num}_block{num_blocks[block_num]-1}_add"
+        f"stack{block_num}_block{num_blocks[block_num] - 1}_add"
     ).output
     return keras.models.Model(
         inputs=x,

{keras_hub_nightly-0.19.0.dev202501080345 → keras_hub_nightly-0.19.0.dev202501150344}/keras_hub/src/models/deeplab_v3/deeplab_v3_layers.py

@@ -88,13 +88,13 @@ class SpatialPyramidPooling(keras.layers.Layer):
                         dilation_rate=dilation_rate,
                         use_bias=False,
                         data_format=self.data_format,
-                        name=f"aspp_conv_{i+2}",
+                        name=f"aspp_conv_{i + 2}",
                     ),
                     keras.layers.BatchNormalization(
-                        axis=self.channel_axis, name=f"aspp_bn_{i+2}"
+                        axis=self.channel_axis, name=f"aspp_bn_{i + 2}"
                     ),
                     keras.layers.Activation(
-                        self.activation, name=f"aspp_activation_{i+2}"
+                        self.activation, name=f"aspp_activation_{i + 2}"
                     ),
                 ]
             )

{keras_hub_nightly-0.19.0.dev202501080345 → keras_hub_nightly-0.19.0.dev202501150344}/keras_hub/src/models/densenet/densenet_backbone.py

@@ -81,14 +81,14 @@ class DenseNetBackbone(FeaturePyramidBackbone):
                 channel_axis,
                 stackwise_num_repeats[stack_index],
                 growth_rate,
-                name=f"stack{stack_index+1}",
+                name=f"stack{stack_index + 1}",
             )
             pyramid_outputs[f"P{index}"] = x
             x = apply_transition_block(
                 x,
                 channel_axis,
                 compression_ratio,
-                name=f"transition{stack_index+1}",
+                name=f"transition{stack_index + 1}",
             )
 
         x = apply_dense_block(
@@ -140,7 +140,7 @@ def apply_dense_block(x, channel_axis, num_repeats, growth_rate, name=None):
 
     for i in range(num_repeats):
         x = apply_conv_block(
-            x, channel_axis, growth_rate, name=f"{name}_block{i+1}"
+            x, channel_axis, growth_rate, name=f"{name}_block{i + 1}"
         )
     return x
 

{keras_hub_nightly-0.19.0.dev202501080345 → keras_hub_nightly-0.19.0.dev202501150344}/keras_hub/src/models/flux/flux_text_to_image.py

@@ -81,7 +81,7 @@ class FluxTextToImage(TextToImage):
 
     def fit(self, *args, **kwargs):
         raise NotImplementedError(
-            "Currently, `fit` is not supported for " "`FluxTextToImage`."
+            "Currently, `fit` is not supported for `FluxTextToImage`."
         )
 
     def generate_step(

{keras_hub_nightly-0.19.0.dev202501080345 → keras_hub_nightly-0.19.0.dev202501150344}/keras_hub/src/models/pali_gemma/pali_gemma_presets.py

@@ -5,7 +5,7 @@ backbone_presets = {
     "pali_gemma_3b_mix_224": {
         "metadata": {
             "description": (
-                "image size 224, mix fine tuned, text sequence " "length is 256"
+                "image size 224, mix fine tuned, text sequence length is 256"
             ),
             "params": 2923335408,
             "path": "pali_gemma",
@@ -45,7 +45,7 @@ backbone_presets = {
     "pali_gemma_3b_896": {
         "metadata": {
             "description": (
-                "image size 896, pre trained, text sequence length " "is 512"
+                "image size 896, pre trained, text sequence length is 512"
             ),
             "params": 2927759088,
             "path": "pali_gemma",

{keras_hub_nightly-0.19.0.dev202501080345 → keras_hub_nightly-0.19.0.dev202501150344}/keras_hub/src/models/resnet/resnet_backbone.py

@@ -177,7 +177,7 @@ class ResNetBackbone(FeaturePyramidBackbone):
                 use_bias=False,
                 padding="same",
                 dtype=dtype,
-                name=f"conv{conv_index+1}_conv",
+                name=f"conv{conv_index + 1}_conv",
             )(x)
 
         if not use_pre_activation:

{keras_hub_nightly-0.19.0.dev202501080345 → keras_hub_nightly-0.19.0.dev202501150344}/keras_hub/src/models/retinanet/feature_pyramid.py

@@ -209,9 +209,9 @@ class FeaturePyramid(keras.layers.Layer):
             )
             if i == backbone_max_level + 1 and self.use_p5:
                 self.output_conv_layers[level].build(
-                    (None, None, None, input_shapes[f"P{i-1}"][-1])
+                    (None, None, None, input_shapes[f"P{i - 1}"][-1])
                     if self.data_format == "channels_last"
-                    else (None, input_shapes[f"P{i-1}"][1], None, None)
+                    else (None, input_shapes[f"P{i - 1}"][1], None, None)
                 )
             else:
                 self.output_conv_layers[level].build(
@@ -277,7 +277,7 @@ class FeaturePyramid(keras.layers.Layer):
             if i < backbone_max_level:
                 # for the top most output, it doesn't need to merge with any
                 # upper stream outputs
-                upstream_output = self.top_down_op(output_features[f"P{i+1}"])
+                upstream_output = self.top_down_op(output_features[f"P{i + 1}"])
                 output = self.merge_op([output, upstream_output])
             output_features[level] = (
                 self.lateral_batch_norm_layers[level](output)
@@ -296,9 +296,9 @@ class FeaturePyramid(keras.layers.Layer):
         for i in range(backbone_max_level + 1, self.max_level + 1):
             level = f"P{i}"
             feats_in = (
-                inputs[f"P{i-1}"]
+                inputs[f"P{i - 1}"]
                 if i == backbone_max_level + 1 and self.use_p5
-                else output_features[f"P{i-1}"]
+                else output_features[f"P{i - 1}"]
             )
             if i > backbone_max_level + 1:
                 feats_in = self.activation(feats_in)