Skip to content

Commit

Permalink
fix entity class name for binary sensor
Browse files Browse the repository at this point in the history
  • Loading branch information
Knapoc authored May 26, 2020
1 parent 4a8d401 commit b4490e9
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions custom_components/nintendo_wishlist/binary_sensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,13 @@
from typing import List

from homeassistant import core
from homeassistant.components.binary_sensor import BinarySensorDevice
from homeassistant.util import slugify

try:
from homeassistant.components.binary_sensor import BinarySensorEntity
except ImportError:
from homeassistant.components.binary_sensor import BinarySensorDevice as BinarySensorEntity

from .const import DOMAIN
from .types import SwitchGame

Expand All @@ -23,7 +27,7 @@ async def async_setup_platform(
async_add_entities(sensors, True)


class SwitchGameQueryEntity(BinarySensorDevice):
class SwitchGameQueryEntity(BinarySensorEntity):
"""Represents a Query for a switch game."""

def __init__(self, coordinator, game_title: str):
Expand Down

0 comments on commit b4490e9

Please sign in to comment.