export Trial and TrialStatus (#948)

keras-team · Sep 26, 2023 · f8a5a45 · f8a5a45
1 parent d8c2f90
commit f8a5a45
Show file tree

Hide file tree

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

diff --git a/keras_tuner/engine/trial.py b/keras_tuner/engine/trial.py
@@ -20,11 +20,13 @@
 
 from keras_tuner import protos
 from keras_tuner import utils
+from keras_tuner.api_export import keras_tuner_export
 from keras_tuner.engine import hyperparameters as hp_module
 from keras_tuner.engine import metrics_tracking
 from keras_tuner.engine import stateful
 
 
+@keras_tuner_export(["keras_tuner.engine.trial.TrialStatus"])
 class TrialStatus:
     # The Trial may start to run.
     RUNNING = "RUNNING"
@@ -83,6 +85,7 @@ def from_proto(proto):
             raise ValueError(f"Unknown status {proto}")
 
 
+@keras_tuner_export(["keras_tuner.engine.trial.Trial"])
 class Trial(stateful.Stateful):
     """The runs with the same set of hyperparameter values.
 

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.1"
+__version__ = "1.4.2"