diff --git a/timecopilot/models/neural.py b/timecopilot/models/neural.py index cc5665d..2276f94 100644 --- a/timecopilot/models/neural.py +++ b/timecopilot/models/neural.py @@ -354,6 +354,8 @@ def forecast( if self.config is None: config = _AutoNBEATS.get_default_config(h=h, backend="ray") config["scaler_type"] = tune.choice(["robust"]) + if h == 1: + config["stack_types"] = ["identity"] else: config = self.config if self.backend == "optuna":