aboutsummaryrefslogtreecommitdiff
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