export Trial and TrialStatus (#948)

Author: haifeng-jin

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

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.
 

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"