diff --git a/dont_fret/web/datamanager.py b/dont_fret/web/datamanager.py index 85662de..4127f9d 100644 --- a/dont_fret/web/datamanager.py +++ b/dont_fret/web/datamanager.py @@ -78,7 +78,7 @@ async def get_bursts( try: photon_data = await self.get_photons(photon_node) - bursts = await self.run(full_search, photon_data, burst_colors, self.cfg) + bursts = await self.run(full_search, photon_data, burst_colors, self.cfg.hooks) future.set_result(bursts) except Exception as e: diff --git a/tests/test_web.py b/tests/test_web.py index 318c03a..44abba9 100644 --- a/tests/test_web.py +++ b/tests/test_web.py @@ -72,7 +72,7 @@ async def test_burst_search(): assert new_node.bursts burst_item = new_node.bursts[0] assert burst_item.name == "DCBS" - assert burst_item.df.shape == (72, 22) + assert burst_item.df.shape == (72, 24) assert burst_item.df["filename"].unique()[0] == "datafile_1.ptu" await asyncio.sleep(0)