@@ -62,7 +62,7 @@ def _ndcg_parallel_numba(qrels, run, k, rel_lvl, jarvelin):
6262 NUMBA_AVAILABLE = False
6363
6464
65- def _dcg_parallel_numpy (qrels , run , k , rel_lvl , jarvelin ):
65+ def _dcg_numpy (qrels , run , k , rel_lvl , jarvelin ):
6666 """NumPy fallback implementation."""
6767 scores = np .zeros ((len (qrels )), dtype = np .float64 )
6868 for i in range (len (qrels )):
@@ -77,7 +77,7 @@ def _dcg_parallel(qrels, run, k, rel_lvl, jarvelin):
7777 if NUMBA_AVAILABLE and use_numba ():
7878 return _dcg_parallel_numba (qrels , run , k , rel_lvl , jarvelin )
7979 else :
80- return _dcg_parallel_numpy (qrels , run , k , rel_lvl , jarvelin )
80+ return _dcg_numpy (qrels , run , k , rel_lvl , jarvelin )
8181
8282
8383@maybe_njit (cache = True )
@@ -97,7 +97,7 @@ def _ndcg(qrels, run, k, rel_lvl, jarvelin):
9797 return dcg_score / idcg_score
9898
9999
100- def _ndcg_parallel_numpy (qrels , run , k , rel_lvl , jarvelin ):
100+ def _ndcg_numpy (qrels , run , k , rel_lvl , jarvelin ):
101101 """NumPy fallback implementation."""
102102 scores = np .zeros ((len (qrels )), dtype = np .float64 )
103103 for i in range (len (qrels )):
@@ -112,7 +112,7 @@ def _ndcg_parallel(qrels, run, k, rel_lvl, jarvelin):
112112 if NUMBA_AVAILABLE and use_numba ():
113113 return _ndcg_parallel_numba (qrels , run , k , rel_lvl , jarvelin )
114114 else :
115- return _ndcg_parallel_numpy (qrels , run , k , rel_lvl , jarvelin )
115+ return _ndcg_numpy (qrels , run , k , rel_lvl , jarvelin )
116116
117117
118118# HIGH LEVEL FUNCTIONS =========================================================
0 commit comments