diff --git a/bigframes/ml/linear_model.py b/bigframes/ml/linear_model.py index 8ee7269361..722b72f806 100644 --- a/bigframes/ml/linear_model.py +++ b/bigframes/ml/linear_model.py @@ -155,7 +155,6 @@ def _fit( def predict(self, X: utils.ArrayType) -> bpd.DataFrame: if not self._bqml_model: raise RuntimeError("A model must be fitted before predict") - # add session (X,) = utils.batch_convert_to_dataframe(X, session=self._bqml_model.session) return self._bqml_model.predict(X)