Update tsa.py

vertica · Nov 8, 2023 · 174e7f4 · 174e7f4
1 parent 214602d
commit 174e7f4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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: