From 4a9fd93d01b6d4bd9d9880dbf2274b3c7f2de37b Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Fri, 6 May 2022 16:43:26 +0300
Subject: Fix linting and typing errors

---
 gn3/db/partial_correlations.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'gn3/db')

diff --git a/gn3/db/partial_correlations.py b/gn3/db/partial_correlations.py
index 96f953d..e8466f6 100644
--- a/gn3/db/partial_correlations.py
+++ b/gn3/db/partial_correlations.py
@@ -9,7 +9,7 @@ This module is part of the optimisation effort for the partial correlations.
 """
 
 from functools import reduce, partial
-from typing import Any, Dict, Tuple, Union, Sequence
+from typing import Any, Dict, Tuple, Union, Sequence, Generator
 
 from MySQLdb.cursors import DictCursor
 
@@ -750,7 +750,7 @@ def traits_datasets(conn, threshold, traits):
 
 def traits_info(
         conn: Any, threshold: int, traits_fullnames: Tuple[str, ...],
-        qtl=None) -> Tuple[Dict[str, Any], ...]:
+        qtl=None) -> Generator:
     """
     Retrieve basic trait information for multiple `traits`.
 
-- 
cgit v1.2.3