improving style on how default values are given to base_estimator_hps

awslabs · Jul 24, 2023 · 45310e7 · 45310e7
1 parent 86af2d8
commit 45310e7
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/gluonts/nursery/temporal_hierarchical_forecasting/model/cop_deepar/_estimator.py b/src/gluonts/nursery/temporal_hierarchical_forecasting/model/cop_deepar/_estimator.py
@@ -207,20 +207,24 @@ def __init__(
  assert self.base_estimator_type == DeepAREstimatorForCOP
 
  if "distr_output" not in base_estimator_hps:
- base_estimator_hps["distr_output"] = GaussianOutput()
+ base_estimator_hps.setdefault("distr_output", GaussianOutput())
 
  print(f"Distribution output: {base_estimator_hps['distr_output']}")
 
  if "impute_missing_values" not in base_estimator_hps:
- base_estimator_hps["impute_missing_values"] = impute_missing_values
+ base_estimator_hps.setdefault(
+ "impute_missing_values", impute_missing_values
+ )
 
  if "imputation_method" not in base_estimator_hps:
- base_estimator_hps["imputation_method"] = imputation_method
+ base_estimator_hps.setdefault(
+ "imputation_method", imputation_method
+ )
 
  if "num_imputation_samples" not in base_estimator_hps:
- base_estimator_hps[
- "num_imputation_samples"
- ] = num_imputation_samples
+ base_estimator_hps.setdefault(
+ "num_imputation_samples", num_imputation_samples
+ )
 
  self.estimators = []
  for agg_multiple, freq_str in zip(