fix import

Author: adriangb

pyproject.toml

@@ -51,6 +51,7 @@ pre-commit = ">=2.20.0"
 pytest = ">=7.1.2"
 pytest-cov = ">=3.0.0"
 sphinx = ">=5.0.2"
+tensorflow = ">=2.16.1"
 
 [tool.ruff]
 select = [

scikeras/_saving_utils.py

@@ -3,9 +3,8 @@
 
 import keras as keras
 import keras.saving
-import keras.saving.object_registration
 import numpy as np
-from keras.saving.saving_lib import load_model, save_model
+from keras.src.saving.saving_lib import load_model, save_model
 
 
 def unpack_keras_model(
@@ -25,7 +24,7 @@ def pack_keras_model(
     """Support for Pythons's Pickle protocol."""
     tp = type(model)
     out = BytesIO()
-    if tp not in keras.saving.object_registration.GLOBAL_CUSTOM_OBJECTS:
+    if tp not in keras.saving.get_custom_objects():
         module = ".".join(tp.__qualname__.split(".")[:-1])
         name = tp.__qualname__.split(".")[-1]
         keras.saving.register_keras_serializable(module, name)(tp)

tests/test_compile_kwargs.py

@@ -5,9 +5,9 @@
 from keras import losses as losses_module
 from keras import metrics as metrics_module
 from keras import optimizers as optimizers_module
-from keras.backend.common.variables import KerasVariable
 from keras.layers import Dense, Input
 from keras.models import Model
+from keras.src.backend.common.variables import KerasVariable
 from sklearn.datasets import make_classification
 
 from scikeras.wrappers import KerasClassifier