Skip to content
Snippets Groups Projects
Commit 5704efd7 authored by pswain's avatar pswain
Browse files

fix(concat_one_signal): ensured kwargs passed to signal.get_raw

parent 7d7eaf10
No related branches found
No related tags found
No related merge requests found
...@@ -291,8 +291,13 @@ def concat_one_signal( ...@@ -291,8 +291,13 @@ def concat_one_signal(
position_name=None, position_name=None,
tmax_in_mins_dict=None, tmax_in_mins_dict=None,
cutoff: float = 0, cutoff: float = 0,
**kwargs,
) -> pd.DataFrame: ) -> 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: if tmax_in_mins_dict and position_name in tmax_in_mins_dict:
tmax_in_mins = tmax_in_mins_dict[position_name] tmax_in_mins = tmax_in_mins_dict[position_name]
else: else:
...@@ -313,10 +318,10 @@ def concat_one_signal( ...@@ -313,10 +318,10 @@ def concat_one_signal(
) )
elif mode == "raw": elif mode == "raw":
# no picking and merging # 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": elif mode == "raw_daughters":
combined = position.get_raw( 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: if combined is not None:
combined = combined.loc[ combined = combined.loc[
...@@ -324,7 +329,7 @@ def concat_one_signal( ...@@ -324,7 +329,7 @@ def concat_one_signal(
] ]
elif mode == "raw_mothers": elif mode == "raw_mothers":
combined = position.get_raw( 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: if combined is not None:
combined = combined.loc[ combined = combined.loc[
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment