fix tests

Jhsmit · Nov 25, 2024 · 5b7c413 · 5b7c413
1 parent b8ccfae
commit 5b7c413
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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)