remove flask g object

1 files changed, 8 insertions, 3 deletions

diff --git a/wqflask/wqflask/correlation/rust_correlation.py b/wqflask/wqflask/correlation/rust_correlation.py
index 2a2ad4a0..ce1b5fda 100644
--- a/wqflask/wqflask/correlation/rust_correlation.py
+++ b/wqflask/wqflask/correlation/rust_correlation.py
@@ -1,7 +1,6 @@
 """module contains integration code for rust-gn3"""
 import json
 from functools import reduce
-from flask import g
 from utility.db_tools import mescape
 from utility.db_tools import create_in_clause
 from wqflask.correlation.correlation_functions import get_trait_symbol_and_tissue_values
@@ -29,8 +28,12 @@ def chunk_dataset(dataset,steps,name):
                   ProbeSetXRef.ProbeSetId = ProbeSet.Id
     """.format(name)
 
-    traits_name_dict = dict(g.db.execute(query).fetchall())
+    with database_connector() as conn:
+        curr = conn.cursor()
 
+        curr.execute(query)
+
+        traits_name_dict = curr.fetchall()
 
     for i in range(0, len(dataset), steps):
         matrix = list(dataset[i:i + steps])
@@ -45,7 +48,7 @@ def chunk_dataset(dataset,steps,name):
 def compute_top_n_sample(start_vars, dataset, trait_list):
     """check if dataset is of type probeset"""
 
-    if dataset.type!= "Probeset":
+    if dataset.type.lower()!= "probeset":
         return  {}
 
     def __fetch_sample_ids__(samples_vals, samples_group):
@@ -248,6 +251,8 @@ def compute_correlation_rust(
 
     top_a = top_b = {}
 
+    compute_all  =  True
+
     if compute_all:
 
         if corr_type == "sample":