Change to a more version forgiving approach for dropping columns

vega · Mar 12, 2024 · 16ccfab · 16ccfab
1 parent 8a549f2
commit 16ccfab
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/examples_arguments_syntax/interactive_column_selection.py b/tests/examples_arguments_syntax/interactive_column_selection.py
@@ -29,7 +29,7 @@
 select_x = alt.selection_point(fields=['level_0'], name='select_x', value='b')
 select_y = alt.selection_point(fields=['level_1'], name='select_y', value='d')
 heatmap = alt.Chart(
-    ex_ts.corr(numeric_only=True).stack().reset_index(name='correlation'),
+    ex_ts.drop(columns='date').corr().stack().reset_index(name='correlation'),
     title='Click a tile to compare timeseries',
     height=250,
     width=250,

diff --git a/tests/examples_methods_syntax/interactive_column_selection.py b/tests/examples_methods_syntax/interactive_column_selection.py
@@ -27,7 +27,7 @@
 select_x = alt.selection_point(fields=['level_0'], name='select_x', value='b')
 select_y = alt.selection_point(fields=['level_1'], name='select_y', value='d')
 heatmap = alt.Chart(
-    ex_ts.corr(numeric_only=True).stack().reset_index(name='correlation'),
+    ex_ts.drop(columns='date').corr().stack().reset_index(name='correlation'),
     title='Click a tile to compare timeseries',
     height=250,
     width=250,