Export BaseTuner (#946)

keras-team · Sep 25, 2023 · d8c2f90 · d8c2f90
1 parent b491a38
commit d8c2f90
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/RELEASE.md b/RELEASE.md
@@ -1,3 +1,8 @@
+# Release v1.4.1
+
+## Bug fixes
+* Could not do `from keras_tuner.engine import base_tuner`. It is now fixed.
+
 # Release v1.4.0
 
 ## Breaking changes

diff --git a/keras_tuner/engine/base_tuner.py b/keras_tuner/engine/base_tuner.py
@@ -23,6 +23,7 @@
 from keras_tuner import config as config_module
 from keras_tuner import errors
 from keras_tuner import utils
+from keras_tuner.api_export import keras_tuner_export
 from keras_tuner.distribute import utils as dist_utils
 from keras_tuner.engine import hypermodel as hm_module
 from keras_tuner.engine import oracle as oracle_module
@@ -31,6 +32,7 @@
 from keras_tuner.engine import tuner_utils
 
 
+@keras_tuner_export(["keras_tuner.engine.base_tuner.BaseTuner"])
 class BaseTuner(stateful.Stateful):
     """Tuner base class.
 

diff --git a/keras_tuner/version.py b/keras_tuner/version.py
@@ -13,4 +13,4 @@
 # limitations under the License.
 
 # Unique source of truth for the version number.
-__version__ = "1.4.0"
+__version__ = "1.4.1"