Fix tests

cwindolf · Nov 22, 2023 · a665f63 · a665f63
1 parent eda817c
commit a665f63
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/dartsort/templates/pairwise_util.py b/src/dartsort/templates/pairwise_util.py
@@ -1031,6 +1031,11 @@ def coarse_approximate(
 
                 active_temp_a = temp_ix_a[inshift]
                 unique_active_temp_a = np.unique(active_temp_a)
+
+                # TODO just upsampling dedup
+                # active_temp_b = temp_ix_b[inshift]
+                # unique_active_temp_b = np.unique(active_temp_b)
+                # if unique_active_temp_a.size == unique_active_temp_b.size == 1:
                 if unique_active_temp_a.size == 1:
                     new_pconv.append(convs)
                     old_ix_to_new_ix[inshift] = np.arange(

diff --git a/src/dartsort/templates/template_util.py b/src/dartsort/templates/template_util.py
@@ -264,6 +264,7 @@ def temporally_upsample_templates(
         "n_compressed_upsampled_templates",
         "compressed_upsampled_templates",
         "compressed_upsampling_map",
+        "compressed_upsampling_index",
         "compressed_index_to_template_index",
         "compressed_index_to_upsampling_index",
     ],
@@ -360,6 +361,7 @@ def compressed_upsampled_templates(
         current_compressed_index,
         compressed_upsampled_templates,
         compressed_upsampling_map,
+        compressed_upsampling_index,
         template_indices,
         upsampling_indices,
     )