Skip to content

Commit

Permalink
Update bigframes/ml/linear_model.py
Browse files Browse the repository at this point in the history
  • Loading branch information
tswast authored Dec 20, 2024
1 parent 0971bf6 commit 1d8a1fd
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion bigframes/ml/linear_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 1d8a1fd

Please sign in to comment.