remove deprecated fields

flyteorg · Oct 10, 2020 · c48d496 · c48d496
1 parent 23aa95c
commit c48d496
Showing 1 changed file with 1 addition and 10 deletions.
diff --git 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