keras-nightly 3.12.0.dev2025090103__py3-none-any.whl → 3.12.0.dev2025090303__py3-none-any.whl

keras/src/backend/jax/layer.py

@@ -3,7 +3,9 @@ from keras.src.backend.config import is_nnx_enabled
 if is_nnx_enabled():
     from flax import nnx
 
-    BaseLayer = nnx.Module
+    class BaseLayer(nnx.Module):
+        def __init_subclass__(cls, **kwargs):
+            super().__init_subclass__(pytree=False, **kwargs)
 else:
     BaseLayer = object
 

keras/src/layers/preprocessing/image_preprocessing/bounding_boxes/validation.py

@@ -7,7 +7,7 @@ def _classes_shape(batched, classes_shape, max_boxes):
         return None
     if batched:
         return [None, max_boxes] + classes_shape[2:]
-    return [max_boxes] + classes_shape[2:]
+    return [max_boxes] + classes_shape[1:]
 
 
 def _box_shape(batched, boxes_shape, max_boxes):
@@ -91,20 +91,20 @@ def densify_bounding_boxes(
                         labels_default_value for _ in range(num_boxes_to_add)
                     ]
             return {
-                "boxes": backend.convert_to_tensor(new_boxes, dtype="int32"),
-                "labels": backend.convert_to_tensor(new_boxes, dtype="int32"),
+                "boxes": backend.convert_to_tensor(new_boxes, dtype="float32"),
+                "labels": backend.convert_to_tensor(new_labels, dtype="int32"),
             }
 
     if tf_utils.is_ragged_tensor(boxes):
         bounding_boxes["boxes"] = bounding_boxes["boxes"].to_tensor(
             default_value=boxes_default_value,
-            shape=_classes_shape(
+            shape=_box_shape(
                 is_batched, bounding_boxes["boxes"].shape, max_boxes
             ),
         )
         bounding_boxes["labels"] = bounding_boxes["labels"].to_tensor(
             default_value=labels_default_value,
-            shape=_box_shape(
+            shape=_classes_shape(
                 is_batched, bounding_boxes["labels"].shape, max_boxes
             ),
         )

keras/src/ops/operation.py

@@ -130,15 +130,55 @@ class Operation(KerasSaveable):
                 vars(instance)["_object__state"] = nnx.object.ObjectState()
 
         # Generate a config to be returned by default by `get_config()`.
-        arg_names = inspect.getfullargspec(cls.__init__).args
-        kwargs.update(dict(zip(arg_names[1 : len(args) + 1], args)))
+        auto_config = True
+
+        signature = inspect.signature(cls.__init__)
+        argspec = inspect.getfullargspec(cls.__init__)
+
+        try:
+            bound_parameters = signature.bind(None, *args, **kwargs)
+        except TypeError:
+            # Raised by signature.bind when the supplied args and kwargs
+            # do not match the signature.
+            auto_config = False
+
+        if auto_config and any(
+            [
+                param.kind == inspect.Parameter.POSITIONAL_ONLY
+                for name, param in signature.parameters.items()
+                if name != argspec.args[0]
+            ]
+        ):
+            # cls.__init__ takes positional only arguments, which
+            # cannot be restored via cls(**config)
+            auto_config = False
+            # Create variable to show appropriate warning in get_config.
+            instance._auto_config_error_args = True
+
+        if auto_config:
+            # Include default values in the config.
+            bound_parameters.apply_defaults()
+            # Extract all arguments as a dictionary.
+            kwargs = bound_parameters.arguments
+            # Expand variable kwargs argument.
+            kwargs |= kwargs.pop(argspec.varkw, {})
+            # Remove first positional argument, self.
+            kwargs.pop(argspec.args[0])
+            # Remove argument "name", as it is provided by get_config.
+            kwargs.pop("name", None)
+            if argspec.varargs is not None:
+                # Varargs cannot be meaningfully converted to a dictionary.
+                varargs = kwargs.pop(argspec.varargs)
+                if len(varargs) > 0:
+                    auto_config = False
+                    # Store variable to show appropriate warning in get_config.
+                    instance._auto_config_error_args = True
 
         # For safety, we only rely on auto-configs for a small set of
         # serializable types.
         supported_types = (str, int, float, bool, type(None))
         try:
             flat_arg_values = tree.flatten(kwargs)
-            auto_config = True
             for value in flat_arg_values:
                 if not isinstance(value, supported_types):
                     auto_config = False
@@ -193,30 +233,52 @@ class Operation(KerasSaveable):
                 config.pop("name", None)
             return config
         else:
-            raise NotImplementedError(
-                textwrap.dedent(
-                    f"""
-        Object {self.__class__.__name__} was created by passing
-        non-serializable argument values in `__init__()`,
-        and therefore the object must override `get_config()` in
-        order to be serializable. Please implement `get_config()`.
-
-        Example:
-
-        class CustomLayer(keras.layers.Layer):
-            def __init__(self, arg1, arg2, **kwargs):
-                super().__init__(**kwargs)
-                self.arg1 = arg1
-                self.arg2 = arg2
-
-            def get_config(self):
-                config = super().get_config()
-                config.update({"arg1": self.arg1,
-                    "arg2": self.arg2,
-                })
-                return config"""
+            example_str = """
+            class CustomLayer(keras.layers.Layer):
+                def __init__(self, arg1, arg2, **kwargs):
+                    super().__init__(**kwargs)
+                    self.arg1 = arg1
+                    self.arg2 = arg2
+
+                def get_config(self):
+                    config = super().get_config()
+                    config.update({
+                        "arg1": self.arg1,
+                        "arg2": self.arg2,
+                    })
+                    return config
+            """
+            if getattr(self, "_auto_config_error_args", False):
+                raise NotImplementedError(
+                    textwrap.dedent(
+                        f"""
+            Object {self.__class__.__name__} was created by passing
+            positional only or variadic positional arguments (e.g.,
+            `*args`) to `__init__()`, which is not supported by the
+            automatic config generation. Please remove all positional
+            only and variadic arguments from `__init__()`
+            or override `get_config()` and `from_config()` to make
+            the object serializatble.
+
+            Example:
+
+            {example_str}"""
+                    )
+                )
+            else:
+                raise NotImplementedError(
+                    textwrap.dedent(
+                        f"""
+            Object {self.__class__.__name__} was created by passing
+            non-serializable argument values in `__init__()`,
+            and therefore the object must override `get_config()` in
+            order to be serializable. Please implement `get_config()`.
+
+            Example:
+
+            {example_str}"""
+                    )
                 )
-            )
 
     @classmethod
     def from_config(cls, config):

keras/src/version.py

@@ -1,7 +1,7 @@
 from keras.src.api_export import keras_export
 
 # Unique source of truth for the version number.
-__version__ = "3.12.0.dev2025090103"
+__version__ = "3.12.0.dev2025090303"
 
 
 @keras_export("keras.version")

{keras_nightly-3.12.0.dev2025090103.dist-info → keras_nightly-3.12.0.dev2025090303.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: keras-nightly
-Version: 3.12.0.dev2025090103
+Version: 3.12.0.dev2025090303
 Summary: Multi-backend Keras
 Author-email: Keras team <keras-users@googlegroups.com>
 License: Apache License 2.0

{keras_nightly-3.12.0.dev2025090103.dist-info → keras_nightly-3.12.0.dev2025090303.dist-info}/RECORD

@@ -126,7 +126,7 @@ keras/regularizers/__init__.py,sha256=542Shphw7W8h4Dyf2rmqMKUECVZ8IVBvN9g1LWhz-b
 keras/saving/__init__.py,sha256=KvL2GZxjvgFgEhvEnkvqjIR9JSNHKz-NWZacXajsjLI,1298
 keras/src/__init__.py,sha256=Gi4S7EiCMkE03PbdGNpFdaUYySWDs_FcAJ8Taz9Y1BE,684
 keras/src/api_export.py,sha256=gXOkBOnmscV013WAc75lc4Up01-Kkg9EylIAT_QWctg,1173
-keras/src/version.py,sha256=wR6UDH2ffMhDy3CqP7QIiRJaoJxgLXXi65Oe-E2PWE4,204
+keras/src/version.py,sha256=10rfmpGBb6KJKrifewkNq8aWHatUOvbKV8SoOP3HJUw,204
 keras/src/activations/__init__.py,sha256=0nL3IFDB9unlrMz8ninKOWo-uCHasTUpTo1tXZb2u44,4433
 keras/src/activations/activations.py,sha256=mogPggtp4CGldI3VOPNmesRxp6EbiR1_i4KLGaVwzL8,17614
 keras/src/applications/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -165,7 +165,7 @@ keras/src/backend/jax/core.py,sha256=kd2-Q3Nm5zp7YwyO_d6ZFlqUCPKP7_hVFAUrgD_FvIU
 keras/src/backend/jax/distribution_lib.py,sha256=-pm5qtFbzTgg_Z7sxq30x4Mjdopn-6qgXFysWoNLtZo,8663
 keras/src/backend/jax/export.py,sha256=jV2yKQLzYjK72vTJmdNomWPLeNS_lDTCEKzQx_5D_-E,7368
 keras/src/backend/jax/image.py,sha256=RiYIalbIaUQdDOGpDZUBk5KNsX94Xqg7iyXGATN9V58,30482
-keras/src/backend/jax/layer.py,sha256=QxZeeiimUulsb3j1h3ncNxIoTYdKPO89s0kP49ZwF-w,194
+keras/src/backend/jax/layer.py,sha256=o6CicT06udwamTRQIjNSDLZLyYHFzBXNbxewXgWe0iw,308
 keras/src/backend/jax/linalg.py,sha256=dtGHRYCvoVlRX0UwbDDdunA8Vp_mA3sdqoasX4P8SbQ,2532
 keras/src/backend/jax/math.py,sha256=1IEDpdoF8e5ltu3D4wbDQuihzvJHhMXz8W9Z_E-eJqU,9391
 keras/src/backend/jax/nn.py,sha256=R0a8-WB0YCl14FpRi2CQ45MFRvHCFtPTedk0Q1LfWYc,45935
@@ -410,7 +410,7 @@ keras/src/layers/preprocessing/image_preprocessing/bounding_boxes/bounding_box.p
 keras/src/layers/preprocessing/image_preprocessing/bounding_boxes/converters.py,sha256=opXlmX5SRiWEU2_M1PqkiBVi8LRNfIIDMPMfMY_2Yp0,15999
 keras/src/layers/preprocessing/image_preprocessing/bounding_boxes/formats.py,sha256=b4v7nskUauUvk7Ub4rgImPUysJrDl4m5oBTGD1MEnTI,3377
 keras/src/layers/preprocessing/image_preprocessing/bounding_boxes/iou.py,sha256=ZwqkN4d2QoYlK6sIv2OJ7MkRyObLmdRMKflb1Tp0YJ4,10140
-keras/src/layers/preprocessing/image_preprocessing/bounding_boxes/validation.py,sha256=aNeC8VBG2xY7qvNAcbTpb824SDyOf88iGMNIohwsjQk,7189
+keras/src/layers/preprocessing/image_preprocessing/bounding_boxes/validation.py,sha256=BIbkLBosoWWXAiqqgeDcg7w7NpP3K9ODXWrv1TYceFo,7192
 keras/src/layers/regularization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras/src/layers/regularization/activity_regularization.py,sha256=FCUAcb3Pyotp2lmoks6Ixb6lmNfg-Pa4eGgW6C7S5zQ,1283
 keras/src/layers/regularization/alpha_dropout.py,sha256=z2Kl3Pblx4eZnJWUiAUgUogu4A1pTPclwzH_Sa-B4Fs,3625
@@ -490,7 +490,7 @@ keras/src/ops/math.py,sha256=4qYMJ5qAPmeSyeF63YWoGbUkQt6f4_VX0enOChU4mXU,37233
 keras/src/ops/nn.py,sha256=1BC-zmnpsUhqG5lSE4VvV5PsBf81wN0ZGg4kU-R8TJY,95259
 keras/src/ops/node.py,sha256=aJgn9D-GkteE--Bbt2cZ9JjVxb2W2uS1OWEKoeLsl3Y,5583
 keras/src/ops/numpy.py,sha256=ovWnm3XHDIzjiwjm_GjM0oVZHCGige7Ek95cX4qCAvk,236694
-keras/src/ops/operation.py,sha256=Q-nOnXPrmt2daaC3X1png3Y86sFPDttrNEopPb6o3wM,12957
+keras/src/ops/operation.py,sha256=dpPI6bQsdBk6j0EUNygoLRHngrMTDoqT2Z55mgq6hbE,15520
 keras/src/ops/operation_utils.py,sha256=BSarr5DZF5dr-URdXNzawwZlFx6R7VRjh6P2DGwgrT4,14457
 keras/src/ops/symbolic_arguments.py,sha256=MKwXxZYkyouD9BPmQ1uUNxILdcwPvTayAqXaUV3P3o4,1628
 keras/src/optimizers/__init__.py,sha256=k7AmJUexCuGHTvU5gCrL_Pf7XYQmA6dZjJ47kcLvqfk,3974
@@ -597,7 +597,7 @@ keras/utils/bounding_boxes/__init__.py,sha256=jtvQll4u8ZY0Z96HwNhP1nxWEG9FM3gI-6
 keras/utils/legacy/__init__.py,sha256=oSYZz6uS8UxSElRaaJYWJEoweJ4GAasZjnn7fNaOlog,342
 keras/visualization/__init__.py,sha256=UKWmiy6sps4SWlmQi9WX8_Z53cPpLlphz2zIeHdwJpQ,722
 keras/wrappers/__init__.py,sha256=QkS-O5K8qGS7C3sytF8MpmO6PasATpNVGF8qtb7Ojsw,407
-keras_nightly-3.12.0.dev2025090103.dist-info/METADATA,sha256=8h4vgmfQG-ojifS5KcReo2WDWMppje7vYId2_3kTNY4,5970
-keras_nightly-3.12.0.dev2025090103.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-keras_nightly-3.12.0.dev2025090103.dist-info/top_level.txt,sha256=ptcw_-QuGZ4ZDjMdwi_Z0clZm8QAqFdvzzFnDEOTs9o,6
-keras_nightly-3.12.0.dev2025090103.dist-info/RECORD,,
+keras_nightly-3.12.0.dev2025090303.dist-info/METADATA,sha256=rA14PbRaPmupzN7_lPZAarq7u9P8Zfkv8Ahlyd8YJWg,5970
+keras_nightly-3.12.0.dev2025090303.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+keras_nightly-3.12.0.dev2025090303.dist-info/top_level.txt,sha256=ptcw_-QuGZ4ZDjMdwi_Z0clZm8QAqFdvzzFnDEOTs9o,6
+keras_nightly-3.12.0.dev2025090303.dist-info/RECORD,,