From 7fbb50a24c393b748579fb30a2ee7ccb673291d4 Mon Sep 17 00:00:00 2001 From: Pedro Eduardo Mercado Lopez Date: Mon, 24 Jul 2023 15:18:26 +0200 Subject: [PATCH] improving style on how default values are given to base_estimator_hps - remove ifs --- .../model/cop_deepar/_estimator.py | 22 +++++++------------ 1 file changed, 8 insertions(+), 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 index 0e12e2aadd..e4acc4cf13 100644 --- 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(