Add edge map parameter to track linearization

CHANGELOG.md

@@ -6,6 +6,11 @@
 
 <!-- Running draft to be removed immediately prior to release. -->
 
+```python
+from spyglass.linearization.v1.main import TrackGraph
+TrackGraph.alter() # Comment regarding the change
+```
+
 ### Infrastructure
 
 - Disable populate transaction protection for long-populating tables #1066
@@ -17,6 +22,8 @@
 - Decoding
 
     - Fix edge case errors in spike time loading #1083
+- Linearization
+    - Add edge_map parameter to LinearizedPositionV1 #1091
 
 - Spike Sorting
 

src/spyglass/linearization/v1/main.py

@@ -49,6 +49,7 @@ class TrackGraph(SpyglassMixin, dj.Manual):
     edges: blob                # shape (n_edges, 2)
     linear_edge_order : blob   # order of edges in linear space, (n_edges, 2)
     linear_edge_spacing : blob # space btwn edges in linear space, (n_edges,)
+    edge_map = NULL : blob     # Maps one edge to another before linearization
     """
 
     def get_networkx_track_graph(self, track_graph_parameters=None):
@@ -58,6 +59,7 @@ def get_networkx_track_graph(self, track_graph_parameters=None):
         return make_track_graph(
             node_positions=track_graph_parameters["node_positions"],
             edges=track_graph_parameters["edges"],
+            edge_map=track_graph_parameters["edge_map"],
         )
 
     def plot_track_graph(self, ax=None, draw_edge_labels=False, **kwds):