From b90db2c1af80cd3520c58478bf7b91f50910023a Mon Sep 17 00:00:00 2001
From: Joe Testa <jtesta@positronsecurity.com>
Date: Fri, 18 Apr 2025 17:06:29 -0400
Subject: [PATCH] Fixed mypy failure.
---
src/ssh_audit/dheat.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/ssh_audit/dheat.py b/src/ssh_audit/dheat.py
index 6c7e99f..3696e47 100644
--- a/src/ssh_audit/dheat.py
+++ b/src/ssh_audit/dheat.py
@@ -759,9 +759,9 @@ class DHEat:
r = socket.getaddrinfo(host, 0, family, socket.SOCK_STREAM)
for address_family, socktype, _, _, addr in r:
if socktype == socket.SOCK_STREAM:
- return address_family, addr[0]
+ return int(address_family), str(addr[0])
- return -1, ''
+ return int(socket.AF_UNSPEC), ''
def _run(self) -> bool: