Skip to content

Commit df2b1d2

Browse files
committed
Final updates and fixes
1 parent 07753f3 commit df2b1d2

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

hybrid-demo/build_verification_test.ipynb

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -541,7 +541,6 @@
541541
"def forecast_revenue(df, start_date, end_date):\n",
542542
" # Filter data from last year\n",
543543
" df_filtered = df[(df[\"DATE\"] >= start_date - pd.Timedelta(days=365)) & (df[\"DATE\"] < start_date)]\n",
544-
" \n",
545544
" # Append future dates to daily_avg for prediction\n",
546545
" future_dates = pd.date_range(start=start_date, end=end_date, freq=\"D\")\n",
547546
" df_future = pd.DataFrame({\"DATE\": future_dates})\n",
@@ -595,8 +594,8 @@
595594
"outputs": [],
596595
"source": [
597596
"# TODO: Waiting on GroupBy switcheroo\n",
598-
"df_forecast = df_forecast.move_to('Pandas')\n",
599-
"assert df_forecast.get_backend() == 'Pandas'"
597+
"#df_forecast = df_forecast.move_to('Pandas')\n",
598+
"assert df_forecast.get_backend() == 'Snowflake'"
600599
]
601600
},
602601
{

src/snowflake/snowpark/modin/plugin/extensions/pd_extensions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def _snowpark_pandas_obj_check(obj: Union[DataFrame, Series]):
3939

4040
@register_pd_accessor("explain")
4141
def explain(last=20) -> native_pd.DataFrame:
42-
stats = get_hybrid_switch_log().tail(20)
42+
stats = get_hybrid_switch_log().tail(last)
4343
stats = stats.drop_duplicates().fillna(value=' ')
4444
stats2 = stats.reset_index()
4545
return stats2.set_index([ 'source', 'mode', 'from', 'to', 'metric']).drop(columns=['index', 'group'])

0 commit comments

Comments
 (0)