From 3b80f956755c9d7043138aab6e5687cba50be8cb Mon Sep 17 00:00:00 2001
From: Garrett Wu <6505921+GarrettWu@users.noreply.github.com>
Date: Tue, 26 Mar 2024 14:09:44 -0700
Subject: [PATCH] docs: fix docs of ARIMAPlus.predict (#512)

---
 bigframes/ml/forecasting.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/bigframes/ml/forecasting.py b/bigframes/ml/forecasting.py
index 292389dcbb..7993327200 100644
--- a/bigframes/ml/forecasting.py
+++ b/bigframes/ml/forecasting.py
@@ -234,7 +234,12 @@ def _fit(
     def predict(
         self, X=None, *, horizon: int = 3, confidence_level: float = 0.95
     ) -> bpd.DataFrame:
-        """Predict the closest cluster for each sample in X.
+        """Forecast time series at future horizon.
+
+        .. note::
+
+            Output matches that of the BigQuery ML.FORECAST function.
+            See: https://cloud.google.com/bigquery/docs/reference/standard-sql/bigqueryml-syntax-forecast
 
         Args:
             X (default None):