Update __init__.py

Fix backward compatibility
hif2k1 · Feb 7, 2023 · db5b0f3 · db5b0f3
1 parent f9e5d27
commit db5b0f3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/custom_components/battery_sim/__init__.py b/custom_components/battery_sim/__init__.py
@@ -64,6 +64,8 @@
         {
             vol.Required(CONF_IMPORT_SENSOR): cv.entity_id,
             vol.Required(CONF_EXPORT_SENSOR): cv.entity_id,
+            vol.Optional(CONF_ENERGY_TARIFF): cv.entity_id,
+            vol.Optional(CONF_ENERGY_EXPORT_TARIFF): cv.entity_id,
             vol.Optional(CONF_ENERGY_IMPORT_TARIFF): cv.entity_id,
             vol.Optional(CONF_NAME): cv.string,
             vol.Required(CONF_BATTERY_SIZE): vol.All(float),