Generics and other type hint improvements (#45250)

This commit is contained in:
Ville Skyttä
2021-01-18 23:23:25 +02:00
committed by GitHub
parent 4928476abe
commit 94dbcc9d2b
7 changed files with 92 additions and 45 deletions

View File

@ -1,6 +1,6 @@
"""Unit system helper class and methods."""
from numbers import Number
from typing import Optional
from typing import Dict, Optional
from homeassistant.const import (
CONF_UNIT_SYSTEM_IMPERIAL,
@ -31,6 +31,8 @@ from homeassistant.util import (
volume as volume_util,
)
# mypy: disallow-any-generics
LENGTH_UNITS = distance_util.VALID_UNITS
MASS_UNITS = [MASS_POUNDS, MASS_OUNCES, MASS_KILOGRAMS, MASS_GRAMS]
@ -135,7 +137,7 @@ class UnitSystem:
# type ignore: https://github.com/python/mypy/issues/7207
return volume_util.convert(volume, from_unit, self.volume_unit) # type: ignore
def as_dict(self) -> dict:
def as_dict(self) -> Dict[str, str]:
"""Convert the unit system to a dictionary."""
return {
LENGTH: self.length_unit,