From 10256824e545d8e4115286389ef38f2d2c9735e1 Mon Sep 17 00:00:00 2001
From: Alexander_Kabui
Date: Thu, 7 Jul 2022 23:13:14 +0300
Subject: mypy pylint fixes

---
 gn3/computations/rust_correlation.py             | 3 ++-
 tests/unit/computations/test_rust_correlation.py | 4 ----
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/gn3/computations/rust_correlation.py b/gn3/computations/rust_correlation.py
index 0cd3ee2..0b37144 100644
--- a/gn3/computations/rust_correlation.py
+++ b/gn3/computations/rust_correlation.py
@@ -31,7 +31,8 @@ def generate_input_files(dataset: list[str],
     return (tmp_dir, tmp_file)
 
 
-def generate_json_file(tmp_dir, tmp_file, method, delimiter, x_vals) -> str:
+def generate_json_file(tmp_dir, tmp_file,
+                       method, delimiter, x_vals) -> tuple[str, str]:
     """generating json input file required by cargo"""
 
     tmp_json_file = os.path.join(tmp_dir, f"{random_string(10)}.json")
diff --git a/tests/unit/computations/test_rust_correlation.py b/tests/unit/computations/test_rust_correlation.py
index 7c284ed..0fd8e94 100644
--- a/tests/unit/computations/test_rust_correlation.py
+++ b/tests/unit/computations/test_rust_correlation.py
@@ -68,9 +68,6 @@ def test_parse_results():
         "num_overlap":  00,
                 "p_value": p_val}} for (trait, corr_coeff, p_val) in raw_data]
 
-    expected_results = [{"trait_name": name, "corr_coeff": corr,
-                         "p_val": pval} for (name, corr, pval) in raw_data]
-
     assert (parse_correlation_output(
         "tests/unit/computations/data/correlation/sorted_results.txt",
         len(raw_data))
@@ -122,7 +119,6 @@ def test_get_samples():
 
     }
 
-    exluded = ["BXD", "BXD11"]
     assert get_samples(all_samples=al_samples,
                        base_samples=["BXD", "BXD4", "BXD5", "BXD6",
                                      "BXD11"
-- 
cgit v1.2.3