Update populate

src/spyglass/common/populate_all_common.py

@@ -9,6 +9,7 @@
 from .common_nwbfile import Nwbfile
 from .common_session import Session
 from .common_task import TaskEpoch
+from spyglass.spikesorting.merge import SpikeSortingOutput
 
 
 def populate_all_common(nwb_file_name):
@@ -49,3 +50,6 @@ def populate_all_common(nwb_file_name):
     print("RawPosition...")
     PositionSource.insert_from_nwbfile(nwb_file_name)
     RawPosition.populate(fp)
+
+    print("Populate SpikeSortingOutput...")
+    SpikeSortingOutput.populate(fp)

src/spyglass/spikesorting/imported.py

@@ -1,5 +1,7 @@
 import datajoint as dj
 from spyglass.common.common_session import Session
+from spyglass.common.common_nwbfile import Nwbfile
+import pynwb
 
 schema = dj.schema("spikesorting_imported")
 
@@ -11,3 +13,16 @@ class ImportedSpikeSorting(dj.Imported):
     ---
     object_id: varchar(32)
     """
+
+    def make(self, key):
+        nwb_file_abs_path = Nwbfile.get_abs_path(key["nwb_file_name"])
+
+        with pynwb.NWBHDF5IO(
+            nwb_file_abs_path, "r", load_namespaces=True
+        ) as io:
+            nwbfile = io.read()
+            if nwbfile.units:
+                key["object_id"] = nwbfile.units.object_id
+                self.insert1(key, skip_duplicates=True)
+            else:
+                print("No units found in NWB file")