Imports

Author: yger

src/spikeinterface/postprocessing/tests/test_template_similarity.py

@@ -9,14 +9,14 @@
 from spikeinterface.postprocessing import check_equal_template_with_distribution_overlap, ComputeTemplateSimilarity
 from spikeinterface.postprocessing.template_similarity import (
     compute_similarity_with_templates_array,
-    _compute_similarity_matrix_numba,
     _compute_similarity_matrix_numpy,
 )
 
 try:
     import numba
 
     HAVE_NUMBA = True
+    from spikeinterface.postprocessing.template_similarity import _compute_similarity_matrix_numba
 except ModuleNotFoundError as err:
     HAVE_NUMBA = False