aboutsummaryrefslogtreecommitdiff
path: root/gn3/computations/datasets.py
diff options
context:
space:
mode:
authorAlexander Kabui2021-04-05 14:34:01 +0300
committerAlexander Kabui2021-04-05 14:34:01 +0300
commita301093ce9ea825606db29bd1f1f40e1971897dd (patch)
tree10f0ee83b127a5cb907eb66dfe9955b44a0cd105 /gn3/computations/datasets.py
parent2301b11e8a975f2e6dc7e5144e4b26c34b186501 (diff)
downloadgenenetwork3-a301093ce9ea825606db29bd1f1f40e1971897dd.tar.gz
fix for fetching dataset traits data
Diffstat (limited to 'gn3/computations/datasets.py')
-rw-r--r--gn3/computations/datasets.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/gn3/computations/datasets.py b/gn3/computations/datasets.py
index b405e55..3664e4a 100644
--- a/gn3/computations/datasets.py
+++ b/gn3/computations/datasets.py
@@ -1,6 +1,5 @@
"""module contains the code all related to datasets"""
import json
-from unittest import mock
from math import ceil
from collections import defaultdict
@@ -16,7 +15,8 @@ from gn3.settings import GN2_BASE_URL
def retrieve_trait_sample_data(dataset,
trait_name: str,
- group_species_id=None,) -> List:
+ database,
+ group_species_id=None) -> List:
"""given the dataset id and trait_name fetch the\
sample_name,value from the dataset"""
@@ -35,7 +35,8 @@ def retrieve_trait_sample_data(dataset,
if dataset_query:
formatted_query = dataset_query % sample_query_values[dataset_type]
- results = fetch_from_db_sample_data(formatted_query, mock.Mock())
+
+ results = fetch_from_db_sample_data(formatted_query, database)
return results