fix

Author: GarrettWu

bigframes/ml/base.py

@@ -248,7 +248,11 @@ def fit(
     ) -> _T:
         return self._fit(X, y)
 
-    def fit_predict(self, X: utils.ArrayType, y=None) -> bpd.DataFrame:  # ignored
+    def fit_predict(
+        self: _T,
+        X: utils.ArrayType,
+        y: Optional[utils.ArrayType] = None,
+    ) -> _T:
         return self.fit(X).predict(X)
 
 

notebooks/generative_ai/bq_dataframes_llm_kmeans.ipynb

@@ -1736,7 +1736,7 @@
    "provenance": []
   },
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": "venv (3.10.14)",
    "language": "python",
    "name": "python3"
   },
@@ -1750,7 +1750,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.9"
+   "version": "3.10.14"
   }
  },
  "nbformat": 4,