|
161 | 161 | }
|
162 | 162 | ],
|
163 | 163 | }
|
164 |
| -STUDY_CONFIG_FIXED_BOOLEAN = { |
165 |
| - "metrics": [{"goal": "MAXIMIZE", "metric": "accuracy"}], |
166 |
| - "parameters": [ |
167 |
| - { |
168 |
| - "discrete_value_spec": {"values": [1.0]}, |
169 |
| - "parameter": "condition", |
170 |
| - "type": "DISCRETE", |
171 |
| - } |
172 |
| - ], |
173 |
| -} |
174 | 164 | OPTIMIZER_TRIAL = {
|
175 | 165 | "name": "projects/project/locations/region/studies/study/trials/trial_1",
|
176 | 166 | "state": "ACTIVE",
|
@@ -283,8 +273,7 @@ def test_convert_study_config_bool(self):
|
283 | 273 |
|
284 | 274 | @parameterized.parameters(
|
285 | 275 | ("beta", 0.1, STUDY_CONFIG_FIXED_FLOAT),
|
286 |
| - ("type", "WIDE_AND_DEEP", STUDY_CONFIG_FIXED_CATEGORICAL), |
287 |
| - ("condition", True, STUDY_CONFIG_FIXED_BOOLEAN)) |
| 276 | + ("type", "WIDE_AND_DEEP", STUDY_CONFIG_FIXED_CATEGORICAL)) |
288 | 277 | def test_convert_study_config_fixed(self, name, value, expected_config):
|
289 | 278 | hps = hp_module.HyperParameters()
|
290 | 279 | hps.Fixed(name, value)
|
@@ -401,15 +390,6 @@ def test_convert_hyperparams_to_hparams_fixed(self, name, value):
|
401 | 390 | }
|
402 | 391 | self.assertEqual(repr(hparams), repr(expected_hparams))
|
403 | 392 |
|
404 |
| - def test_convert_hyperparams_to_hparams_fixed_bool(self): |
405 |
| - hps = hp_module.HyperParameters() |
406 |
| - hps.Fixed("condition", True) |
407 |
| - hparams = utils.convert_hyperparams_to_hparams(hps) |
408 |
| - expected_hparams = { |
409 |
| - hparams_api.HParam("condition", hparams_api.Discrete([1])): 1, |
410 |
| - } |
411 |
| - self.assertEqual(repr(hparams), repr(expected_hparams)) |
412 |
| - |
413 | 393 | @parameterized.parameters(
|
414 | 394 | ("val_loss", "min",
|
415 | 395 | [oracle_module.Objective(name="val_loss", direction="min")]),
|
|
0 commit comments