Miscellaneous ESPHome cleanups

homeassistant/components/binary_sensor/esphome.py

@@ -60,4 +60,4 @@ def available(self):
         """Return True if entity is available."""
         if self._static_info.is_status_binary_sensor:
             return True
-        return super(EsphomeEntity, self).available
+        return super().available

homeassistant/components/esphome/__init__.py

@@ -43,11 +43,11 @@
 
 # The HA component types this integration supports
 HA_COMPONENTS = [
-    'sensor',
     'binary_sensor',
     'cover',
     'fan',
     'light',
+    'sensor',
     'switch',
 ]
 
@@ -461,6 +461,7 @@ async def async_will_remove_from_hass(self):
         """Unregister callbacks."""
         for remove_callback in self._remove_callbacks:
             remove_callback()
+        self._remove_callbacks = []
 
     @property
     def _entry_data(self) -> RuntimeEntryData:

homeassistant/components/switch/esphome.py

@@ -47,6 +47,11 @@ def icon(self) -> str:
         """Return the icon."""
         return self._static_info.icon
 
+    @property
+    def assumed_state(self) -> bool:
+        """Return true if we do optimistic updates."""
+        return self._static_info.optimistic
+
     @property
     def is_on(self):
         """Return true if the switch is on."""