Fix unit of measurement usage in COSignal (#75856)

This commit is contained in:
Franck Nijhof 2022-07-28 12:39:10 +02:00 committed by GitHub
parent 25d943d272
commit 4f56028491
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -103,8 +103,8 @@ class CO2Sensor(CoordinatorEntity[CO2SignalCoordinator], SensorEntity):
@property
def native_unit_of_measurement(self) -> str | None:
"""Return the unit of measurement."""
if self.entity_description.unit_of_measurement:
return self.entity_description.unit_of_measurement
if self.entity_description.native_unit_of_measurement:
return self.entity_description.native_unit_of_measurement
return cast(
str, self.coordinator.data["units"].get(self.entity_description.key)
)