Fix linting and typing errors

1 files changed, 2 insertions, 2 deletions

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`.