Commit 4b651244 authored by JOSSOUD Olivier's avatar JOSSOUD Olivier
Browse files

Flow. get_arrival_df(). Allow custom pickle file name.

parent 76f72e00
Pipeline #152677 passed with stages
in 55 seconds
......@@ -284,7 +284,8 @@ def get_arrival_df(encoder_df: pd.DataFrame,
with_conduct: bool = True,
parallel: bool = True,
force_update: bool = False,
dataset: DatasetReader = None) -> pd.DataFrame:
dataset: DatasetReader = None,
pkl_name: str = None) -> pd.DataFrame:
"""Get arrival datetime of each encoder measure, at collector's needle and Picarro
Parameters
......@@ -314,7 +315,9 @@ def get_arrival_df(encoder_df: pd.DataFrame,
Dataframe with arrival datetime at needle and Picarro.
"""
if dataset is not None:
pkl_filename = os.path.join(dataset.binary_base_path, "arrival.pkl")
if pkl_name is None:
pkl_name = "arrival.pkl"
pkl_filename = os.path.join(dataset.binary_base_path, pkl_name)
else:
pkl_filename = "----"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment