PyPI - keras-hub-nightly - Versions diffs - 0.19.0.dev202501090358__tar.gz → 0.19.0.dev202501160344__tar.gz - Mend

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: keras-hub-nightly
-Version: 0.19.0.dev202501090358
+Version: 0.19.0.dev202501160344
 Summary: Industry-strength Natural Language Processing extensions for Keras.
 Home-page: https://github.com/keras-team/keras-hub
 Author: Keras team

@@ -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)

@@ -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,

@@ -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}"
                     ),
                 ]
             )

@@ -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

@@ -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(

@@ -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",

@@ -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:

@@ -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)

@@ -351,7 +351,7 @@ class ViTEncoder(keras.layers.Layer):
                 attention_dropout=self.attention_dropout,
                 layer_norm_epsilon=self.layer_norm_epsilon,
                 dtype=self.dtype_policy,
-                name=f"tranformer_block_{i+1}",
+                name=f"tranformer_block_{i + 1}",
             )
             encoder_block.build((None, None, self.hidden_dim))
             self.encoder_layers.append(encoder_block)

@@ -150,8 +150,7 @@ class ByteTokenizer(tokenizer.Tokenizer):
     ):
         if not is_int_dtype(dtype):
             raise ValueError(
-                "Output dtype must be an integer type. "
-                f"Received: dtype={dtype}"
+                f"Output dtype must be an integer type. Received: dtype={dtype}"
             )
         # Check normalization_form.

@@ -1,5 +1,7 @@
 import io
+from keras_hub.src.utils.tensor_utils import assert_tf_libs_installed
 try:
     import sentencepiece as spm
     import tensorflow as tf
@@ -77,6 +79,7 @@ def compute_sentence_piece_proto(
     tf.Tensor([ 4  8 12  5  9 14  5  6 13  4  7 10 11  6 13],
     shape=(15,), dtype=int32)
     """
+    assert_tf_libs_installed("compute_sentence_piece_proto")
     if spm is None:
         raise ImportError(

@@ -1,5 +1,6 @@
 from keras_hub.src.api_export import keras_hub_export
 from keras_hub.src.tokenizers.word_piece_tokenizer import pretokenize
+from keras_hub.src.utils.tensor_utils import assert_tf_libs_installed
 try:
     import tensorflow as tf
@@ -117,6 +118,8 @@ def compute_word_piece_vocabulary(
     inputs.map(tokenizer.tokenize)
     ```
     """  # noqa: E501
+    assert_tf_libs_installed("compute_word_piece_vocabulary")
     # Read data files.
     if not isinstance(data, (list, tf.data.Dataset)):
         raise ValueError(

@@ -59,9 +59,11 @@ def convert_weights(backbone, loader, timm_config):
     num_stacks = len(backbone.stackwise_num_repeats)
     for stack_index in range(num_stacks):
         for block_idx in range(backbone.stackwise_num_repeats[stack_index]):
-            keras_name = f"stack{stack_index+1}_block{block_idx+1}"
+            keras_name = f"stack{stack_index + 1}_block{block_idx + 1}"
             hf_name = (
-                f"features.denseblock{stack_index+1}.denselayer{block_idx+1}"
+                "features."
+                f"denseblock{stack_index + 1}"
+                f".denselayer{block_idx + 1}"
             )
             port_batch_normalization(f"{keras_name}_1_bn", f"{hf_name}.norm1")
             port_conv2d(f"{keras_name}_1_conv", f"{hf_name}.conv1")
@@ -69,8 +71,8 @@ def convert_weights(backbone, loader, timm_config):
             port_conv2d(f"{keras_name}_2_conv", f"{hf_name}.conv2")
     for stack_index in range(num_stacks - 1):
-        keras_transition_name = f"transition{stack_index+1}"
-        hf_transition_name = f"features.transition{stack_index+1}"
+        keras_transition_name = f"transition{stack_index + 1}"
+        hf_transition_name = f"features.transition{stack_index + 1}"
         port_batch_normalization(
             f"{keras_transition_name}_bn", f"{hf_transition_name}.norm"
         )

@@ -268,7 +268,7 @@ def convert_weights(backbone, loader, timm_config):
             # 97 is the start of the lowercase alphabet.
             letter_identifier = chr(block_idx + 97)
-            keras_block_prefix = f"block{stack_index+1}{letter_identifier}_"
+            keras_block_prefix = f"block{stack_index + 1}{letter_identifier}_"
             hf_block_prefix = f"blocks.{stack_index}.{block_idx}."
             if block_type == "v1":

@@ -89,7 +89,7 @@ def convert_weights(backbone, loader, timm_config):
         for block_idx in range(backbone.stackwise_num_blocks[stack_index]):
             if version == "v1":
                 keras_name = f"stack{stack_index}_block{block_idx}"
-                hf_name = f"layer{stack_index+1}.{block_idx}"
+                hf_name = f"layer{stack_index + 1}.{block_idx}"
             else:
                 keras_name = f"stack{stack_index}_block{block_idx}"
                 hf_name = f"stages.{stack_index}.blocks.{block_idx}"

@@ -1,7 +1,7 @@
 from keras_hub.src.api_export import keras_hub_export
 # Unique source of truth for the version number.
-__version__ = "0.19.0.dev202501090358"
+__version__ = "0.19.0.dev202501160344"
 @keras_hub_export("keras_hub.version")

@@ -23,7 +23,7 @@ def get_version(rel_path):
 HERE = pathlib.Path(__file__).parent
 README = (HERE / "README.md").read_text()
-VERSION = "0.19.0.dev202501090358"  # get_version("keras_hub/src/version_utils.py")
+VERSION = "0.19.0.dev202501160344"  # get_version("keras_hub/src/version_utils.py")
 setup(
     name="keras-hub-nightly",  # "keras-hub",

keras-hub-nightly 0.19.0.dev202501090358__tar.gz → 0.19.0.dev202501160344__tar.gz

keras-hub-nightly 0.19.0.dev202501090358tar.gz → 0.19.0.dev202501160344tar.gz