Don't expect pandas to be imported under altair

vega · Aug 11, 2024 · fd72c88 · fd72c88
1 parent 0491d5e
commit fd72c88
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/python/vegafusion/tests/altair_mocks/bar/diverging_stacked/mock.py b/python/vegafusion/tests/altair_mocks/bar/diverging_stacked/mock.py
@@ -1,8 +1,9 @@
 # https://altair-viz.github.io/gallery/diverging_stacked_bar_chart.html
 
 import altair as alt
+import pandas as pd
 
-source = alt.pd.DataFrame([
+source = pd.DataFrame([
     {
         "question": "Question 1",
         "type": "Strongly disagree",

diff --git a/python/vegafusion/tests/altair_mocks/casestudy/falkensee/mock.py b/python/vegafusion/tests/altair_mocks/casestudy/falkensee/mock.py
@@ -1,6 +1,7 @@
 # https://altair-viz.github.io/gallery/falkensee.html
 
 import altair as alt
+import pandas as pd
 
 source = [
     {"year": "1875", "population": 1309},
@@ -55,8 +56,8 @@
     }]
 
 
-source = alt.pd.DataFrame(source)
-source2 = alt.pd.DataFrame(source2)
+source = pd.DataFrame(source)
+source2 = pd.DataFrame(source2)
 
 
 line = alt.Chart(source).mark_line(color='#333').encode(