corr: ccc uses use_ari_numba=True by default

use_ari_numba=True should be safe in GTEx
greenelab · Sep 7, 2023 · dc17ff9 · dc17ff9
1 parent 32890b0
commit dc17ff9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/libs/ccc/corr.py b/libs/ccc/corr.py
@@ -78,7 +78,9 @@ def mic(data: pd.DataFrame, estimator="mic_approx", n_jobs=None) -> pd.DataFrame
     )
 
 
-def ccc(data: pd.DataFrame, internal_n_clusters=None, n_jobs=1) -> pd.DataFrame:
+def ccc(
+    data: pd.DataFrame, internal_n_clusters=None, n_jobs=1, use_ari_numba=True
+) -> pd.DataFrame:
     """
     Compute the Clustermatch Correlation Coefficient (CCC).
     """
@@ -89,6 +91,7 @@ def ccc(data: pd.DataFrame, internal_n_clusters=None, n_jobs=1) -> pd.DataFrame:
         data.to_numpy(),
         internal_n_clusters=internal_n_clusters,
         n_jobs=n_jobs,
+        use_ari_numba=use_ari_numba,
     )
 
     corr_mat = squareform(corr_mat)