WIP: Add V1 draft

src/spyglass/spikesorting/v0/spikesorting_burst.py

@@ -1,9 +1,10 @@
-from itertools import combinations
+from itertools import permutations
 from typing import Dict, List, Tuple
 
 import datajoint as dj
 import matplotlib.pyplot as plt
 import numpy as np
+from datajoint.expression import QueryExpression
 from scipy import stats
 from spikeinterface.postprocessing.correlograms import (
     WaveformExtractor,
@@ -315,7 +316,7 @@ def make(self, key) -> None:
         ccgs, bins = self._compute_correlograms(key, params)
 
         unit_pairs = []
-        for u1, u2 in combinations(units, 2):
+        for u1, u2 in permutations(units, 2):
             unit_pairs.append(
                 {
                     **key,
@@ -400,7 +401,7 @@ def plot_by_sort_group_ids(self, key, sort_group_ids=None):
             plt.suptitle(f"sort group {sg_id}", fontsize=20)
 
     def _validate_pair(
-        self, query: dj.QueryExpression, unit1: int, unit2: int
+        self, query: QueryExpression, unit1: int, unit2: int
     ) -> Tuple[int, int]:
         """Ensure that unit1, unit2 is a valid pair in the table.