From c48d4969301d123cd901fd09f668130bdc0d7b7c Mon Sep 17 00:00:00 2001 From: Haytham AbuelFutuh Date: Sat, 10 Oct 2020 13:54:00 +0200 Subject: [PATCH] remove deprecated fields --- tests/flytekit/unit/models/sagemaker/test_hpo_job.py | 11 +---------- 1 file changed, 1 insertion(+), 10 deletions(-) diff --git a/tests/flytekit/unit/models/sagemaker/test_hpo_job.py b/tests/flytekit/unit/models/sagemaker/test_hpo_job.py index d426933100..eab5f209b4 100644 --- a/tests/flytekit/unit/models/sagemaker/test_hpo_job.py +++ b/tests/flytekit/unit/models/sagemaker/test_hpo_job.py @@ -1,4 +1,4 @@ -from flytekit.models.sagemaker import hpo_job, parameter_ranges, training_job +from flytekit.models.sagemaker import hpo_job, training_job def test_hyperparameter_tuning_objective(): @@ -12,14 +12,6 @@ def test_hyperparameter_tuning_objective(): def test_hyperparameter_job_config(): jc = hpo_job.HyperparameterTuningJobConfig( - hyperparameter_ranges=parameter_ranges.ParameterRanges( - parameter_range_map={ - "a": parameter_ranges.CategoricalParameterRange(values=["1", "2"]), - "b": parameter_ranges.IntegerParameterRange( - min_value=0, max_value=10, scaling_type=parameter_ranges.HyperparameterScalingType.LINEAR - ), - } - ), tuning_strategy=hpo_job.HyperparameterTuningStrategy.BAYESIAN, tuning_objective=hpo_job.HyperparameterTuningObjective( objective_type=hpo_job.HyperparameterTuningObjectiveType.MAXIMIZE, metric_name="test_metric" @@ -28,7 +20,6 @@ def test_hyperparameter_job_config(): ) jc2 = hpo_job.HyperparameterTuningJobConfig.from_flyte_idl(jc.to_flyte_idl()) - assert jc2.hyperparameter_ranges == jc.hyperparameter_ranges assert jc2.tuning_strategy == jc.tuning_strategy assert jc2.tuning_objective == jc.tuning_objective assert jc2.training_job_early_stopping_type == jc.training_job_early_stopping_type