Speedup concordance_cc() and pearson_cc() (#42)

audeering · Apr 25, 2023 · 23fdfc5 · 23fdfc5
1 parent 3faf18f
commit 23fdfc5
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/audmetric/core/api.py b/audmetric/core/api.py
@@ -106,8 +106,10 @@ def concordance_cc(
     """
     assert_equal_length(truth, prediction)
 
-    prediction = np.array(list(prediction))
-    truth = np.array(list(truth))
+    if not isinstance(truth, np.ndarray):
+        truth = np.array(list(truth))
+    if not isinstance(prediction, np.ndarray):
+        prediction = np.array(list(prediction))
 
     if len(prediction) < 2:
         return np.NaN
@@ -672,8 +674,10 @@ def pearson_cc(
     """
     assert_equal_length(truth, prediction)
 
-    prediction = np.array(list(prediction))
-    truth = np.array(list(truth))
+    if not isinstance(truth, np.ndarray):
+        truth = np.array(list(truth))
+    if not isinstance(prediction, np.ndarray):
+        prediction = np.array(list(prediction))
 
     if len(prediction) < 2 or prediction.std() == 0:
         return np.NaN