Make column renaming compatible with more pandas versions

vega · Mar 12, 2024 · 53c4401 · 53c4401
1 parent 16ccfab
commit 53c4401
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.drop(columns='date').corr().stack().reset_index(name='correlation'),
+    ex_ts.drop(columns='date').corr().stack().reset_index().rename(columns={0: '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.drop(columns='date').corr().stack().reset_index(name='correlation'),
+    ex_ts.drop(columns='date').corr().stack().reset_index().rename(columns={0: 'correlation'}),
     title='Click a tile to compare timeseries',
     height=250,
     width=250,