From d9d953ffe215ddb2cf5274d4d73b2e04f78a9945 Mon Sep 17 00:00:00 2001
From: AzulGarza <fede.garza.ramirez@gmail.com>
Date: Mon, 27 Apr 2026 18:01:12 -0300
Subject: [PATCH] fix: handle stack types if h is 1

---
 timecopilot/models/neural.py | 2 ++
 1 file changed, 2 insertions(+)

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":