From 239c9422263ca3dc9d886ffd0a17e375d9dd5c67 Mon Sep 17 00:00:00 2001
From: pswain <peter.swain@ed.ac.uk>
Date: Tue, 6 Aug 2024 19:04:11 +0100
Subject: [PATCH] change: added total_squared to bud fluorescence

---
 src/wela/dataloader.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/wela/dataloader.py b/src/wela/dataloader.py
index 76ce136..85a6679 100644
--- a/src/wela/dataloader.py
+++ b/src/wela/dataloader.py
@@ -189,7 +189,7 @@ class dataloader:
         self.a2g_dict = {v: k for (k, v) in self.g2a_dict.items()}
 
     def include_bud_fluorescence(self, grouper, dataname):
-        """Add mean and median bud fluorescence to the h5 files."""
+        """Add statistics for bud fluorescence to the h5 files."""
         # find fluorescence channels
         channels = list(grouper.channels)
         channels.remove("Brightfield")
@@ -199,6 +199,8 @@ class dataloader:
                 [
                     f"/extraction/{channel}/max/median",
                     f"/extraction/{channel}/max/mean",
+                    f"/extraction/{channel}/max/total",
+                    f"/extraction/{channel}/max/total_squared",
                 ]
                 for channel in channels
             ]
-- 
GitLab