Fix import error

Author: kdorr

mplaltair/__init__.py

@@ -1,4 +1,4 @@
-import mplaltair.parse_chart as parse_chart
+import mplaltair.parse_chart
 import matplotlib
 import altair
 import matplotlib.pyplot as plt
@@ -23,7 +23,7 @@ def convert(alt_chart):
     ax : matplotlib.axes
 
     """
-    chart = parse_chart.ChartMetadata(alt_chart)
+    chart = mplaltair.parse_chart.ChartMetadata(alt_chart)
     fig, ax = plt.subplots()
 
     if chart.mark in ['point', 'circle', 'square']:  # scatter

mplaltair/_axis.py

@@ -1,7 +1,7 @@
 import matplotlib.dates as mdates
 import matplotlib.ticker as ticker
 import numpy as np
-from ._data import _locate_channel_data, _locate_channel_dtype, _locate_channel_scale, _locate_channel_axis, _convert_to_mpl_date
+from ._data import _convert_to_mpl_date
 
 
 def _set_limits(channel, mark, ax):
@@ -170,11 +170,7 @@ def _set_tick_formatter(channel, ax):
     Matplotlib only supports datetime.strftime formatting for dates.
     """
     current_axis = {'x': ax.xaxis, 'y': ax.yaxis}
-    format_str = ''
-
-    # format_str = channel.axis.get('format', '')
-    if 'format' in channel.axis:
-        format_str = channel.axis['format']
+    format_str = channel.axis.get('format', '')
 
     if channel.type == 'temporal':
         if not format_str:

mplaltair/_convert.py

@@ -1,5 +1,3 @@
-from ._data import _locate_channel_data, _locate_channel_dtype
-
 
 def _allowed_ranged_marks(enc_channel, mark):
     """TODO: DOCS

mplaltair/tests/test_axis.py

@@ -4,7 +4,7 @@
 import pandas as pd
 from mplaltair import convert
 from .._axis import convert_axis
-from parse_chart import ChartMetadata
+from ..parse_chart import ChartMetadata
 import pytest
 
 df_quant = pd.DataFrame({