diff --git a/mlrun/model.py b/mlrun/model.py index c3851bd462d..e0e9243d8f4 100644 --- a/mlrun/model.py +++ b/mlrun/model.py @@ -656,22 +656,22 @@ class HyperParamOptions(ModelObj): """Hyper Parameter Options Parameters: - param_file (str): hyper params input file path/url, instead of inline - strategy (str): hyper param strategy - grid, list or random - selector (str): selection criteria for best result ([min|max.]), e.g. max.accuracy - stop_condition (str): early stop condition e.g. "accuracy > 0.9" - parallel_runs (int): number of param combinations to run in parallel (over Dask) - dask_cluster_uri (str): db uri for a deployed dask cluster function, e.g. db://myproject/dask - max_iterations (int): max number of runs (in random strategy) - max_errors (int): max number of child runs errors for the overall job to fail - teardown_dask (bool): kill the dask cluster pods after the runs + param_file (str): hyper params input file path/url, instead of inline + strategy (HyperParamStrategies): hyper param strategy - grid, list or random + selector (str): selection criteria for best result ([min|max.]), e.g. max.accuracy + stop_condition (str): early stop condition e.g. "accuracy > 0.9" + parallel_runs (int): number of param combinations to run in parallel (over Dask) + dask_cluster_uri (str): db uri for a deployed dask cluster function, e.g. db://myproject/dask + max_iterations (int): max number of runs (in random strategy) + max_errors (int): max number of child runs errors for the overall job to fail + teardown_dask (bool): kill the dask cluster pods after the runs """ def __init__( self, param_file=None, - strategy=None, - selector: HyperParamStrategies = None, + strategy: typing.Optional[HyperParamStrategies] = None, + selector=None, stop_condition=None, parallel_runs=None, dask_cluster_uri=None, diff --git a/mlrun/runtimes/serving.py b/mlrun/runtimes/serving.py index 45d477acb5e..c2435fd8cf5 100644 --- a/mlrun/runtimes/serving.py +++ b/mlrun/runtimes/serving.py @@ -638,6 +638,9 @@ def deploy( self.spec.secret_sources = self._secrets.to_serial() if self._spec.function_refs: + # ensure the function is available to the UI while deploying the child functions + self.save(versioned=False) + # deploy child functions self._add_ref_triggers() self._deploy_function_refs()