From 44b4f14375bd575ee04db6692d6e6ae0f579e3bd Mon Sep 17 00:00:00 2001
From: Anton Verburg <47820045+antonverburg@users.noreply.github.com>
Date: Sat, 18 Apr 2020 16:01:49 +0200
Subject: [PATCH] Pi4ioe5v9xxxx from ToggleEntity to SwitchDevice (#34369)

---
 homeassistant/components/pi4ioe5v9xxxx/switch.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/homeassistant/components/pi4ioe5v9xxxx/switch.py b/homeassistant/components/pi4ioe5v9xxxx/switch.py
index 5a889cfc318..a7c1e19074c 100644
--- a/homeassistant/components/pi4ioe5v9xxxx/switch.py
+++ b/homeassistant/components/pi4ioe5v9xxxx/switch.py
@@ -4,10 +4,9 @@ import logging
 from pi4ioe5v9xxxx import pi4ioe5v9xxxx  # pylint: disable=import-error
 import voluptuous as vol
 
-from homeassistant.components.switch import PLATFORM_SCHEMA
+from homeassistant.components.switch import PLATFORM_SCHEMA, SwitchDevice
 from homeassistant.const import DEVICE_DEFAULT_NAME
 import homeassistant.helpers.config_validation as cv
-from homeassistant.helpers.entity import ToggleEntity
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -52,7 +51,7 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
     add_entities(switches)
 
 
-class Pi4ioe5v9Switch(ToggleEntity):
+class Pi4ioe5v9Switch(SwitchDevice):
     """Representation of a  pi4ioe5v9 IO expansion IO."""
 
     def __init__(self, name, pin, invert_logic):