diff --git a/neuralprophet/configure.py b/neuralprophet/configure.py index 78f34b1f8..c0bb6bd83 100644 --- a/neuralprophet/configure.py +++ b/neuralprophet/configure.py @@ -22,7 +22,6 @@ @dataclass class Model: - lagged_reg_layers: Optional[List[int]] quantiles: Optional[List[float]] = None def setup_quantiles(self): diff --git a/neuralprophet/df_utils.py b/neuralprophet/df_utils.py index 79157d508..dc183d798 100644 --- a/neuralprophet/df_utils.py +++ b/neuralprophet/df_utils.py @@ -203,7 +203,7 @@ def data_params_definition( norm_type=normalize, ) - if config_lagged_regressors.regressors is not None: + if config_lagged_regressors is not None and config_lagged_regressors.regressors is not None: for covar in config_lagged_regressors.regressors.keys(): if covar not in df.columns: raise ValueError(f"Lagged regressor {covar} not found in DataFrame.")