about summary refs log tree commit diff
diff options
context:
space:
mode:
-rw-r--r--gn3/computations/correlations.py2
-rw-r--r--tests/unit/computations/test_correlation.py1
2 files changed, 1 insertions, 2 deletions
diff --git a/gn3/computations/correlations.py b/gn3/computations/correlations.py
index 8e8bea5..d805af7 100644
--- a/gn3/computations/correlations.py
+++ b/gn3/computations/correlations.py
@@ -240,7 +240,7 @@ def fetch_lit_correlation_data(
         conn,
         input_mouse_gene_id: Optional[str],
         gene_id: str,
-        mouse_gene_id: Optional[str] = None) -> Tuple[str, float]:
+        mouse_gene_id: Optional[str] = None) -> Tuple[str, Optional[float]]:
     """Given input trait mouse gene id and mouse gene id fetch the lit
     corr_data
 
diff --git a/tests/unit/computations/test_correlation.py b/tests/unit/computations/test_correlation.py
index 167b322..11b0fae 100644
--- a/tests/unit/computations/test_correlation.py
+++ b/tests/unit/computations/test_correlation.py
@@ -1,5 +1,4 @@
 """Module contains the tests for correlation"""
-import math
 from unittest import TestCase
 from unittest import mock
 from collections import namedtuple