From d684af0625456fb28eaa2e3c075ae8f16c872857 Mon Sep 17 00:00:00 2001
From: Kyu Hyun Lee <kyuhyun9056@gmail.com>
Date: Mon, 22 Jan 2024 19:02:50 -0800
Subject: [PATCH] Add UnitSelection

---
 src/spyglass/spikesorting/merge.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/spyglass/spikesorting/merge.py b/src/spyglass/spikesorting/merge.py
index 12baefd34..21683a3a1 100644
--- a/src/spyglass/spikesorting/merge.py
+++ b/src/spyglass/spikesorting/merge.py
@@ -114,3 +114,14 @@ def get_firing_rate(cls, key, time, multiunit=False):
             ],
             axis=1,
         )
+
+
+@schema
+class UnitSelection(SpyglassMixin, dj.Manual):
+    definition = """
+    # Units to be selected for downstream analysis
+    -> SpikeSortingOutput
+    unit_group_name : varchar(30)
+    ---
+    unit_group : blob
+    """