utils: fix type hint in is_private_ip()

stratosphereips · Dec 11, 2024 · d78e85c · d78e85c
1 parent 523b947
commit d78e85c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/slips_files/common/slips_utils.py b/slips_files/common/slips_utils.py
@@ -394,7 +394,9 @@ def is_port_in_use(self, port: int) -> bool:
         sock.close()
         return True
 
-    def is_private_ip(self, ip: Union[ipaddress, str]) -> bool:
+    def is_private_ip(
+        self, ip: Union[ipaddress.IPv4Address, ipaddress.IPv6Address, str]
+    ) -> bool:
         ip_classes = {ipaddress.IPv4Address, ipaddress.IPv6Address}
         for class_ in ip_classes:
             if isinstance(ip, class_):