Merge pull request #341 from JaerongA/datajoint_pipeline

fix: 🐛 fix KeyError in BlockDetection
SainsburyWellcomeCentre · Mar 20, 2024 · 6dbb136 · 6dbb136
2 parents cefffb5 + 6fd1d84
commit 6dbb136
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/aeon/dj_pipeline/analysis/block_analysis.py b/aeon/dj_pipeline/analysis/block_analysis.py
@@ -8,8 +8,10 @@
 from matplotlib import path as mpl_path
 
 from aeon.analysis import utils as analysis_utils
-from aeon.dj_pipeline import acquisition, fetch_stream, get_schema_name, streams, tracking
-from aeon.dj_pipeline.analysis.visit import filter_out_maintenance_periods, get_maintenance_periods
+from aeon.dj_pipeline import (acquisition, fetch_stream, get_schema_name,
+                              streams, tracking)
+from aeon.dj_pipeline.analysis.visit import (filter_out_maintenance_periods,
+                                             get_maintenance_periods)
 
 schema = dj.schema(get_schema_name("block_analysis"))
 logger = dj.logger
@@ -516,7 +518,7 @@ def make(self, key):
         )
 
         block_query = acquisition.Environment.BlockState & chunk_restriction
-        block_df = fetch_stream(block_query)[previous_block_start:chunk_end]
+        block_df = fetch_stream(block_query).sort_index()[previous_block_start:chunk_end]
 
         block_ends = block_df[block_df.pellet_ct.diff() < 0]