From 5704efd7941b5c46714ee8665396a853ff79d9e1 Mon Sep 17 00:00:00 2001
From: pswain <peter.swain@ed.ac.uk>
Date: Tue, 16 Jan 2024 15:50:46 +0000
Subject: [PATCH] fix(concat_one_signal): ensured kwargs passed to
signal.get_raw
---
src/postprocessor/grouper.py | 13 +++++++++----
1 file changed, 9 insertions(+), 4 deletions(-)
diff --git a/src/postprocessor/grouper.py b/src/postprocessor/grouper.py
index 7cf22be..e31d533 100644
--- a/src/postprocessor/grouper.py
+++ b/src/postprocessor/grouper.py
@@ -291,8 +291,13 @@ def concat_one_signal(
position_name=None,
tmax_in_mins_dict=None,
cutoff: float = 0,
+ **kwargs,
) -> pd.DataFrame:
- """Retrieve a signal for one position."""
+ """
+ Retrieve a signal for one position.
+
+ kwargs passed to signal.get_raw.
+ """
if tmax_in_mins_dict and position_name in tmax_in_mins_dict:
tmax_in_mins = tmax_in_mins_dict[position_name]
else:
@@ -313,10 +318,10 @@ def concat_one_signal(
)
elif mode == "raw":
# no picking and merging
- combined = position.get_raw(path, tmax_in_mins=tmax_in_mins)
+ combined = position.get_raw(path, tmax_in_mins=tmax_in_mins, **kwargs)
elif mode == "raw_daughters":
combined = position.get_raw(
- path, lineage=True, tmax_in_mins=tmax_in_mins
+ path, lineage=True, tmax_in_mins=tmax_in_mins, **kwargs
)
if combined is not None:
combined = combined.loc[
@@ -324,7 +329,7 @@ def concat_one_signal(
]
elif mode == "raw_mothers":
combined = position.get_raw(
- path, lineage=True, tmax_in_mins=tmax_in_mins
+ path, lineage=True, tmax_in_mins=tmax_in_mins, **kwargs
)
if combined is not None:
combined = combined.loc[
--
GitLab