From 174e7f40806953020760a0cd1b72174afd0886dc Mon Sep 17 00:00:00 2001
From: Badr <badr.ouali@outlook.fr>
Date: Wed, 8 Nov 2023 17:54:32 -0500
Subject: [PATCH] Update tsa.py

---
 verticapy/machine_learning/vertica/tsa.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/verticapy/machine_learning/vertica/tsa.py b/verticapy/machine_learning/vertica/tsa.py
index c13e2dfa9..a37914fef 100755
--- a/verticapy/machine_learning/vertica/tsa.py
+++ b/verticapy/machine_learning/vertica/tsa.py
@@ -470,6 +470,7 @@ def predict(
         vDataFrame
             a new object.
         """
+        ar_ma = False
         if self._model_type in (
             "AR",
             "MA",
@@ -480,6 +481,7 @@ def predict(
                 ts = self.ts
             if isinstance(y, NoneType):
                 y = self.y
+            ar_ma = True
         if isinstance(start, (int, float)):
             start_predict = int(start + 1)
         else:
@@ -505,7 +507,7 @@ def predict(
                     j = j + start
             elif not (isinstance(start, NoneType)):
                 j = start
-            if (output_standard_errors or output_estimated_ts):
+            if (output_standard_errors or output_estimated_ts) and not (ar_ma):
                 if not (output_standard_errors):
                     stde_out = ""
                 else: