ensure tuples are returned from goupby

Jhsmit · Nov 25, 2024 · 8e8bcc3 · 8e8bcc3
1 parent 5b7c413
commit 8e8bcc3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dont_fret/channel_kde.py b/dont_fret/channel_kde.py
@@ -42,8 +42,8 @@ def compute_fret_2cde(
 
     matching_indices = np.intersect1d(df_DA["burst_index"].unique(), df_DD["burst_index"].unique())
 
-    DA_groups = {k: v for k, v in df_DA.group_by("burst_index")}
-    DD_groups = {k: v for k, v in df_DD.group_by("burst_index")}
+    DA_groups = {k: v for k, v in df_DA.group_by(["burst_index"])}
+    DD_groups = {k: v for k, v in df_DD.group_by(["burst_index"])}
 
     N: int = burst_photons["burst_index"].max() + 1  # type: ignore
     output = np.full(N, fill_value=np.nan)