From 31d9aeb1bf4686500406a809d0c5fa4b629125b0 Mon Sep 17 00:00:00 2001
From: Alexander Kabui
Date: Tue, 30 Mar 2021 18:45:13 +0300
Subject: modify getting sample data from db

---
 gn3/computations/traits.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'gn3')

diff --git a/gn3/computations/traits.py b/gn3/computations/traits.py
index 7386745..bbb3073 100644
--- a/gn3/computations/traits.py
+++ b/gn3/computations/traits.py
@@ -1,4 +1,5 @@
 """module contains all operating related to traits"""
+from gn3.computations.datasets import retrieve_trait_sample_data
 
 
 def compute_sum(rhs_val: int, lhs_val: int) -> int:
@@ -28,7 +29,8 @@ def get_trait_sample_data(trait_dataset, trait_name) -> dict:
     try to fetch from the traits dataset redis is only  used for\
     temp dataset type which is not used in this case """
 
-    sample_results = trait_dataset.retrieve_sample_data(trait_name)
+    sample_results = retrieve_trait_sample_data(
+        trait_dataset.id, trait_dataset.type, trait_name)
 
     trait_data = {}
 
-- 
cgit v1.2.3