diff --git a/homeassistant/components/aftership/sensor.py b/homeassistant/components/aftership/sensor.py index c41e5aec7b5..ff5edc92ba0 100644 --- a/homeassistant/components/aftership/sensor.py +++ b/homeassistant/components/aftership/sensor.py @@ -2,6 +2,7 @@ from datetime import timedelta import logging +from pyaftership.tracker import Tracking import voluptuous as vol from homeassistant.components.sensor import PLATFORM_SCHEMA @@ -11,6 +12,7 @@ import homeassistant.helpers.config_validation as cv from homeassistant.helpers.dispatcher import async_dispatcher_send from homeassistant.helpers.entity import Entity from homeassistant.util import Throttle + from .const import DOMAIN _LOGGER = logging.getLogger(__name__) @@ -56,8 +58,6 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( async def async_setup_platform(hass, config, async_add_entities, discovery_info=None): """Set up the AfterShip sensor platform.""" - from pyaftership.tracker import Tracking - apikey = config[CONF_API_KEY] name = config[CONF_NAME]