From afdc022c1f59c1060be159953fdfbaf2e7bea39b Mon Sep 17 00:00:00 2001 From: Paulus Schoutsen Date: Wed, 26 Sep 2018 18:06:44 +0200 Subject: [PATCH] Fix typing --- homeassistant/util/network.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/homeassistant/util/network.py b/homeassistant/util/network.py index b35a8063f..48840f339 100644 --- a/homeassistant/util/network.py +++ b/homeassistant/util/network.py @@ -1,6 +1,6 @@ """Network utilities.""" -from ipaddress import ip_address, ip_network - +from ipaddress import IPv4Address, IPv6Address, ip_address, ip_network +from typing import Union # IP addresses of loopback interfaces LOCAL_IPS = ( @@ -16,7 +16,7 @@ LOCAL_NETWORKS = ( ) -def is_local(address: ip_address) -> bool: +def is_local(address: Union[IPv4Address, IPv6Address]) -> bool: """Check if an address is local.""" return address in LOCAL_IPS or \ any(address in network for network in LOCAL_NETWORKS)