about summary refs log tree commit diff
path: root/gn3/computations
diff options
context:
space:
mode:
Diffstat (limited to 'gn3/computations')
-rw-r--r--gn3/computations/partial_correlations.py27
1 files changed, 27 insertions, 0 deletions
diff --git a/gn3/computations/partial_correlations.py b/gn3/computations/partial_correlations.py
index e73edfd..4ba2ba4 100644
--- a/gn3/computations/partial_correlations.py
+++ b/gn3/computations/partial_correlations.py
@@ -157,3 +157,30 @@ def batch_computed_tissue_correlation(
     `web.webqtl.correlation.correlationFunction.batchCalTissueCorr`"""
     raise Exception("Not implemented!")
     return ({}, {})
+
+def correlations_of_all_tissue_traits(
+        primary_trait_symbol_value_dict: dict, symbol_value_dict: dict,
+        method: str) -> Tuple[dict, dict]:
+    """
+    Computes and returns the correlation of all tissue traits.
+
+    This is a migration of the
+    `web.webqtl.correlation.correlationFunction.calculateCorrOfAllTissueTrait`
+    function in GeneNetwork1.
+    """
+    # The section below existed in the original function, but with the migration
+    # and the proposed rework (in the near future), the values from the database
+    # should be passed into this function, rather than have the function fetch
+    # the data for itself.
+    # ---------------------------------------------------
+    # primary_trait_symbol_value_dict = fetch_gene_symbol_tissue_value_dict_for_trait(
+    #     (trait_symbol,), probeset_freeze_id, conn)
+    # primary_trait_values = primary_trait_symbol_value_dict.vlaues()[0]
+    # symbol_value_dict = fetch_gene_symbol_tissue_value_dict_for_trait(
+    #     tuple(), probeset_freeze_id, conn)
+    # ---------------------------------------------------
+    # We might end up actually getting rid of this function all together as the
+    # rework is done.
+    primary_trait_values = primary_trait_symbol_value_dict.values()[0]
+    return batch_computed_tissue_correlation(
+        primary_trait_values, symbol_value_dict, method)