From 3073a26a612af1d6cba93c92474e75ec83553697 Mon Sep 17 00:00:00 2001
From: AlexanderKabui
Date: Sun, 13 Nov 2022 21:40:25 +0300
Subject: add check for dataset type

---
 wqflask/wqflask/correlation/rust_correlation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wqflask/wqflask/correlation/rust_correlation.py b/wqflask/wqflask/correlation/rust_correlation.py
index 935ab699..f5ee9403 100644
--- a/wqflask/wqflask/correlation/rust_correlation.py
+++ b/wqflask/wqflask/correlation/rust_correlation.py
@@ -23,7 +23,7 @@ from wqflask.correlation.exceptions import WrongCorrelationType
 def query_probes_metadata(dataset, trait_list):
     """query traits metadata in bulk for probeset"""
 
-    if not bool(trait_list):
+    if not bool(trait_list) or dataset.type!="ProbeSet":
         return []
 
     with database_connector() as conn:
-- 
cgit 1.4.1