Change lineages to lineage for save_trk

vanvalenlab · Aug 11, 2021 · 9c1e47f · 9c1e47f
1 parent f099af8
commit 9c1e47f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/deepcell_tracking/tracking.py b/deepcell_tracking/tracking.py
@@ -749,7 +749,7 @@ def dump(self, filename, track_review_dict=None):
         filename = str(filename)
 
         save_trk(filename=filename,
-                 lineages=track_review_dict['tracks'],
+                 lineage=track_review_dict['tracks'],
                  raw=track_review_dict['X'],
                  tracked=track_review_dict['y_tracked'])
 

diff --git a/deepcell_tracking/utils.py b/deepcell_tracking/utils.py
@@ -226,7 +226,7 @@ def save_trks(filename, lineages, raw, tracked):
                     lineage_name='lineages.json')
 
 
-def save_trk(filename, lineages, raw, tracked):
+def save_trk(filename, lineage, raw, tracked):
     """Saves raw, tracked, and lineage data for one movie into a trk_file.
 
     Args:
@@ -245,15 +245,15 @@ def save_trk(filename, lineages, raw, tracked):
         raise ValueError('filename must end with `.trk`. Found %s' % filename)
 
     # Check that lineages is a dictionary or list of length 1
-    if isinstance(lineages, list):
-        if len(lineages) > 1:
+    if isinstance(lineage, list):
+        if len(lineage) > 1:
             raise ValueError('For trk file, lineages must be a dictionary '
                              'or list with a single dictionary')
         else:
-            lineages = lineages[0]
+            lineage = lineage[0]
 
     save_track_data(filename=filename,
-                    lineages=lineages,
+                    lineages=lineage,
                     raw=raw,
                     tracked=tracked,
                     lineage_name='lineage.json')