fixed crash when tabulating cross-correlation function

johannesulf · May 9, 2022 · 3592bf0 · 3592bf0
1 parent 40e4afd
commit 3592bf0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup
 
 setup(name='tabcorr',
-      version='0.8.0',
+      version='0.8.1',
       description='Tabulated correlation functions for halotools',
       url='https://github.com/johannesulf/TabCorr',
       author='Johannes U. Lange',

diff --git a/tabcorr/tabcorr.py b/tabcorr/tabcorr.py
@@ -65,8 +65,8 @@ def compute_tpcf_matrix(mode, pos, tpcf, period, tpcf_args, tpcf_kwargs,
         else:
             i = task
             if len(pos[i]) > 0:
-                n_tot += len(pos[i_1]) * len(pos[i_2])
-                input_queue.put(task)
+                n_tot += len(pos[i])
+                input_queue.put(i)
 
     if verbose:
         pbar = tqdm.tqdm(