Export more APIs for AutoKeras (#954)

Author: haifeng-jin

RELEASE.md

@@ -1,3 +1,10 @@
+# Release v1.4.4
+
+## Bug fixes
+* Could not do `from keras_tuner.engine.hyperparameters import serialize`. It is now fixed.
+* Could not do `from keras_tuner.engine.hyperparameters import deserialize`. It is now fixed.
+* Could not do `from keras_tuner.engine.tuner import maybe_distribute`. It is now fixed.
+
 # Release v1.4.3
 
 ## Bug fixes

keras_tuner/engine/hyperparameters/__init__.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 from keras_tuner import utils
+from keras_tuner.api_export import keras_tuner_export
 from keras_tuner.backend import keras
 from keras_tuner.engine.hyperparameters import hp_types
 from keras_tuner.engine.hyperparameters.hp_types import Boolean
@@ -31,9 +32,11 @@
 ALL_CLASSES = {cls.__name__: cls for cls in OBJECTS}
 
 
+@keras_tuner_export("keras_tuner.engine.hyperparameters.deserialize")
 def deserialize(config):
     return utils.deserialize_keras_object(config, module_objects=ALL_CLASSES)
 
 
+@keras_tuner_export("keras_tuner.engine.hyperparameters.serialize")
 def serialize(obj):
     return utils.serialize_keras_object(obj)

keras_tuner/engine/tuner.py

@@ -465,6 +465,7 @@ def maybe_compute_model_size(model):
     return 0
 
 
+@keras_tuner_export("keras_tuner.engine.tuner.maybe_distribute")
 @contextlib.contextmanager
 def maybe_distribute(distribution_strategy):
     """Distributes if distribution_strategy is set."""

keras_tuner/version.py

@@ -13,4 +13,4 @@
 # limitations under the License.
 
 # Unique source of truth for the version number.
-__version__ = "1.4.3"
+__version__ = "1.4.4.dev0"