aboutsummaryrefslogtreecommitdiff
path: root/gn3/computations/heatmap.py
diff options
context:
space:
mode:
Diffstat (limited to 'gn3/computations/heatmap.py')
-rw-r--r--gn3/computations/heatmap.py7
1 files changed, 5 insertions, 2 deletions
diff --git a/gn3/computations/heatmap.py b/gn3/computations/heatmap.py
index 98ea26f..1b7dfb7 100644
--- a/gn3/computations/heatmap.py
+++ b/gn3/computations/heatmap.py
@@ -3,12 +3,15 @@ This module will contain functions to be used in computation of the data used to
generate various kinds of heatmaps.
"""
+from functools import reduce
+from typing import Any, Dict, Sequence
from gn3.computations.slink import slink
+from gn3.db.traits import retrieve_trait_data, retrieve_trait_info
from gn3.computations.correlations2 import compute_correlation
def export_trait_data(
- trait_data: dict, strainlist: Sequence[str], dtype: str="val",
- var_exists: bool=False, n_exists: bool=False):
+ trait_data: dict, strainlist: Sequence[str], dtype: str = "val",
+ var_exists: bool = False, n_exists: bool = False):
"""
Export data according to `strainlist`. Mostly used in calculating
correlations.