Update main_objs.py

HCDigitalScholarship · Apr 1, 2024 · 301a8e4 · 301a8e4
1 parent 8d14a79
commit 301a8e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/crim_intervals/main_objs.py b/crim_intervals/main_objs.py
@@ -629,7 +629,7 @@ def _ptype_ema_helper(self, row, ngrams):
         for idx, col in zip(offsets, voices):
             filtered_df = ngrams.loc[idx, col]
         filtered_df = filtered_df.dropna(how='all')
-        filtered_df = filtered_df.groupby(level='First').tail(1)
+        filtered_df = filtered_df.groupby(level=0).tail(1)
         emas = self.emaAddresses(df=filtered_df, mode='')
         complete_ema = self.combineEmaAddresses(emas)
         return complete_ema