We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2a29309 commit 9362838Copy full SHA for 9362838
bigframes/ml/core.py
@@ -95,7 +95,17 @@ def _apply_ml_tvf(
95
)
96
97
result_sql = apply_sql_tvf(input_sql)
98
- df = self._session.read_gbq(result_sql, index_col=index_col_ids)
+ df = self._session.read_gbq_query(
99
+ result_sql,
100
+ index_col=index_col_ids,
101
+ # Many ML methods use nested JSON, which isn't yet compatible with
102
+ # joining local results. Also, there is a chance that the results
103
+ # are greater than 10 GB.
104
+ # TODO(b/395912450): Once the limitations with local data are
105
+ # resolved, consider setting allow_large_results only when expected
106
+ # data size is large.
107
+ allow_large_results=True,
108
+ )
109
if df._has_index:
110
df.index.names = index_labels
111
# Restore column labels
0 commit comments