From bb3f6c6cd9b000f6c8738c3dd4433deed71623a1 Mon Sep 17 00:00:00 2001
From: pswain <peter.swain@ed.ac.uk>
Date: Thu, 23 Nov 2023 14:20:39 +0000
Subject: [PATCH] bug(dataloader): a2g_dict not re-created when user changes
 g2a_dict

---
 add_bud_fluorescence.py | 2 +-
 dataloader.py           | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/add_bud_fluorescence.py b/add_bud_fluorescence.py
index 929c5df..5c0a4a9 100644
--- a/add_bud_fluorescence.py
+++ b/add_bud_fluorescence.py
@@ -41,6 +41,6 @@ def add_bud_fluorescence(h5dirpath, signals=["/extraction/GFP/max/median"]):
                 del f["/postprocessing"]
             if "modifiers" in f:
                 del f["/modifiers"]
-            # rerun postprocessor
+            # re-run postprocessor
             pp = PostProcessor(filepath, new_params)
             pp.run()
diff --git a/dataloader.py b/dataloader.py
index eb09578..edbb7f8 100644
--- a/dataloader.py
+++ b/dataloader.py
@@ -227,6 +227,7 @@ class dataloader:
                 self.g2a_dict = {**self.g2a_dict, **extra_g2a_dict}
             elif extra_g2a_dict:
                 self.g2a_dict = extra_g2a_dict
+            self.a2g_dict = {v: k for (k, v) in self.g2a_dict.items()}
             # create instance of grouper
             grouper = self.get_grouper(dataname)
             print("\n---\n" + dataname + "\n---")
-- 
GitLab