googleapis
diff --git a/Collapse file
‎bigframes/bigquery/_operations/ml.py‎
Copy file name to clipboardExpand all lines: bigframes/bigquery/_operations/ml.py
+4Lines changed: 4 additions & 0 deletions b/Collapse file
‎bigframes/bigquery/_operations/ml.py‎
Copy file name to clipboardExpand all lines: bigframes/bigquery/_operations/ml.py
+4Lines changed: 4 additions & 0 deletions
diff --git a/Collapse file
‎bigframes/ml/core.py‎
Copy file name to clipboardExpand all lines: bigframes/ml/core.py
+3-2Lines changed: 3 additions & 2 deletions b/Collapse file
‎bigframes/ml/core.py‎
Copy file name to clipboardExpand all lines: bigframes/ml/core.py
+3-2Lines changed: 3 additions & 2 deletions
@@ -39,6 +39,10 @@ def _to_sql(df_or_sql: Union[pd.DataFrame, dataframe.DataFrame, str]) -> str:
     else:
         bf_df = cast(dataframe.DataFrame, df_or_sql)
 
+    # Cache dataframes to make sure base table is not a snapshot.
+    # Cached dataframe creates a full copy, never uses snapshot.
+    # This is a workaround for internal issue b/310266666.
+    bf_df.cache()
     sql, _, _ = bf_df._to_sql_query(include_index=False)
     return sql
 
 
@@ -436,8 +436,9 @@ def create_model(
         Returns: a BqmlModel, wrapping a trained model in BigQuery
         """
         options = dict(options)
-        # Cache dataframes to make sure base table is not a snapshot
-        # cached dataframe creates a full copy, never uses snapshot
+        # Cache dataframes to make sure base table is not a snapshot.
+        # Cached dataframe creates a full copy, never uses snapshot.
+        # This is a workaround for internal issue b/310266666.
         if y_train is None:
             input_data = X_train.reset_index(drop=True).cache()
         else: