From 0f43309c37f994a90c4c70ea430558a40b2aae43 Mon Sep 17 00:00:00 2001
From: Eric Denovellis <edeno@bu.edu>
Date: Thu, 5 Sep 2024 13:56:26 -0700
Subject: [PATCH 1/3] Add edge map parameter

---
 src/spyglass/linearization/v1/main.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/spyglass/linearization/v1/main.py b/src/spyglass/linearization/v1/main.py
index 76ec85aaf..ff6d51733 100644
--- a/src/spyglass/linearization/v1/main.py
+++ b/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):

From b809ef7e795d5b67261b524d145ebd78a1fd0700 Mon Sep 17 00:00:00 2001
From: Eric Denovellis <edeno@bu.edu>
Date: Thu, 5 Sep 2024 14:01:14 -0700
Subject: [PATCH 2/3] Update CHANGELOG.md

---
 CHANGELOG.md | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index eb98147e8..ee4067663 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -6,6 +6,11 @@
 
 <!-- Running draft to be removed immediately prior to release. -->
 
+```python
+from spyglass.example import Table
+Table.alter() # Comment regarding the change
+```
+
 ### Infrastructure
 
 - Disable populate transaction protection for long-populating tables #1066
@@ -16,6 +21,8 @@
 
 - Decoding
     - Fix edge case errors in spike time loading #1083
+- Linearization
+    - Add edge_map parameter to LinearizedPositionV1 #1091
 
 ## [0.5.3] (August 27, 2024)
 

From 68323e0e8bb92591cc7397c1fbf85d8da8d59d08 Mon Sep 17 00:00:00 2001
From: Eric Denovellis <edeno@bu.edu>
Date: Thu, 5 Sep 2024 14:02:27 -0700
Subject: [PATCH 3/3] Fix alter tables

---
 CHANGELOG.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index ee4067663..3835df3a7 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -7,8 +7,8 @@
 <!-- Running draft to be removed immediately prior to release. -->
 
 ```python
-from spyglass.example import Table
-Table.alter() # Comment regarding the change
+from spyglass.linearization.v1.main import TrackGraph
+TrackGraph.alter() # Comment regarding the change
 ```
 
 ### Infrastructure