improving style on how default values are given to base_estimator_hps…

… - remove ifs
awslabs · Jul 24, 2023 · 7fbb50a · 7fbb50a
1 parent 45310e7
commit 7fbb50a
Showing 1 changed file with 8 additions and 14 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
@@ -206,25 +206,19 @@ def __init__(
 
  assert self.base_estimator_type == DeepAREstimatorForCOP
 
- if "distr_output" not in base_estimator_hps:
- base_estimator_hps.setdefault("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.setdefault(
- "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.setdefault(
- "imputation_method", imputation_method
- )
+ base_estimator_hps.setdefault("imputation_method", imputation_method)
 
- if "num_imputation_samples" not in base_estimator_hps:
- base_estimator_hps.setdefault(
- "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(