fix: Resolve grouping column warning in backwards compatible manner

open2c · Apr 6, 2024 · fe33a9d · fe33a9d
1 parent 744dc33
commit fe33a9d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bioframe/extras.py b/bioframe/extras.py
@@ -305,11 +305,11 @@ def _each(chrom_group):
         seq = str(seq[:])
         gc = []
         for _, bin in chrom_group.iterrows():
-            s = seq[bin.start : bin.end]
+            s = seq[bin["start"] : bin["end"]]
             gc.append(seq_gc(s, mapped_only=mapped_only))
         return gc
 
-    agg = df.groupby("chrom", sort=False).apply(_each, include_groups=False)
+    agg = df.groupby("chrom", sort=False)[["start", "end"]].apply(_each)
     out_col = pd.Series(data=np.concatenate(agg.values), index=df.index).rename("GC")
 
     if return_input: