Update release version and comment orbax checkpoint (#21934)

Author: sachinprasadhs

keras/api/_tf_keras/keras/callbacks/__init__.py

@@ -19,9 +19,6 @@
 from keras.src.callbacks.model_checkpoint import (
     ModelCheckpoint as ModelCheckpoint,
 )
-from keras.src.callbacks.orbax_checkpoint import (
-    OrbaxCheckpoint as OrbaxCheckpoint,
-)
 from keras.src.callbacks.progbar_logger import ProgbarLogger as ProgbarLogger
 from keras.src.callbacks.reduce_lr_on_plateau import (
     ReduceLROnPlateau as ReduceLROnPlateau,

keras/api/callbacks/__init__.py

@@ -19,9 +19,6 @@
 from keras.src.callbacks.model_checkpoint import (
     ModelCheckpoint as ModelCheckpoint,
 )
-from keras.src.callbacks.orbax_checkpoint import (
-    OrbaxCheckpoint as OrbaxCheckpoint,
-)
 from keras.src.callbacks.progbar_logger import ProgbarLogger as ProgbarLogger
 from keras.src.callbacks.reduce_lr_on_plateau import (
     ReduceLROnPlateau as ReduceLROnPlateau,

keras/src/callbacks/orbax_checkpoint.py

@@ -4,7 +4,6 @@
 
 from keras.src import backend
 from keras.src import tree
-from keras.src.api_export import keras_export
 from keras.src.callbacks.monitor_callback import (
     MonitorCallback,  # For metric monitoring logic
 )
@@ -54,7 +53,6 @@ def convert_scalars(obj):
         return state_tree
 
 
-@keras_export("keras.callbacks.OrbaxCheckpoint")
 class OrbaxCheckpoint(MonitorCallback):
     """Callback to save and load model state using Orbax with a similar API to
     ModelCheckpoint.

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"
+__version__ = "3.13.0"
 
 
 @keras_export("keras.version")