Merge branch 'main' of https://github.com/xing-lab-pitt/LiveCellTrack…

…er-dev
xing-lab-pitt · Nov 11, 2023 · 5a7bc89 · 5a7bc89
2 parents 78b39ef + 6bf07ac
commit 5a7bc89
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/livecellx/track/classify_utils.py b/livecellx/track/classify_utils.py
@@ -24,6 +24,9 @@ def load_class2samples_from_json_dir(
         sample_paths = glob.glob(str(sample_json_dir / subfolder / "*.json"))
         for sample_path in sample_paths:
             sample = SingleCellStatic.load_single_cells_json(sample_path)
+            for sc in sample:
+                sc.meta["sample_src_dir"] = str(sample_json_dir)
+                sc.meta["sample_src_class"] = str(subfolder)
             class2samples[subfolder].append(sample)
     return class2samples