aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--wqflask/wqflask/correlation/rust_correlation.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/wqflask/wqflask/correlation/rust_correlation.py b/wqflask/wqflask/correlation/rust_correlation.py
index 7d796e70..10513563 100644
--- a/wqflask/wqflask/correlation/rust_correlation.py
+++ b/wqflask/wqflask/correlation/rust_correlation.py
@@ -229,7 +229,7 @@ def __compute_lit_corr__(
def compute_correlation_rust(
start_vars: dict, corr_type: str, method: str = "pearson",
- n_top: int = 500, compute_all: bool = False):
+ n_top: int = 500, should_compute_all: bool = False):
"""function to compute correlation"""
target_trait_info = create_target_this_trait(start_vars)
(this_dataset, this_trait, target_dataset, sample_data) = (
@@ -248,7 +248,7 @@ def compute_correlation_rust(
top_a = top_b = {}
- if compute_all:
+ if should_compute_all:
if corr_type == "sample":