Correct keyerror exception. (#52150)

2021-06-26 20:09:53 +02:00 · 2021-06-26 20:09:53 +02:00 · c558c77413
commit c558c77413
parent 5687ced7b3
1 changed files with 2 additions and 2 deletions
--- a/homeassistant/components/modbus/validators.py
+++ b/homeassistant/components/modbus/validators.py
@ -40,10 +40,10 @@ def sensor_schema_validator(config):
            structure = (
                f">{DEFAULT_STRUCT_FORMAT[config[CONF_DATA_TYPE]][config[CONF_COUNT]]}"
            )
-        except KeyError:
+        except KeyError as key:
            raise vol.Invalid(
                f"Unable to detect data type for {config[CONF_NAME]} sensor, try a custom type"
-            ) from KeyError
+            ) from key
    else:
        structure = config.get(CONF_STRUCTURE)