From 5719c962d5f278915f87f4b505b6c8e41c64e5b4 Mon Sep 17 00:00:00 2001 From: qian-chu Date: Tue, 22 Oct 2024 02:38:48 +0200 Subject: [PATCH] Update preprocess.py --- pyneon/preprocess/preprocess.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/pyneon/preprocess/preprocess.py b/pyneon/preprocess/preprocess.py index fa318b8..3379a46 100644 --- a/pyneon/preprocess/preprocess.py +++ b/pyneon/preprocess/preprocess.py @@ -317,13 +317,13 @@ def concat_streams( concat_data = pd.DataFrame(index=new_ts) for stream in stream_info["stream"]: - resamp_data = stream.interpolate( + interp_data = stream.interpolate( new_ts, interp_float_kind, interp_other_kind, inplace=inplace ).data - assert concat_data.shape[0] == resamp_data.shape[0] - assert concat_data.index.equals(resamp_data.index) - concat_data = pd.concat([concat_data, resamp_data], axis=1) - assert concat_data.shape[0] == resamp_data.shape[0] + assert concat_data.shape[0] == interp_data.shape[0] + assert concat_data.index.equals(interp_data.index) + concat_data = pd.concat([concat_data, interp_data], axis=1) + assert concat_data.shape[0] == interp_data.shape[0] return concat_data