From 6a244afadd9edc517cc3ad0e1d22f5c30f67ecef Mon Sep 17 00:00:00 2001 From: Titov Leonid Date: Thu, 5 Dec 2024 13:27:54 +0300 Subject: [PATCH] =?UTF-8?q?fix:=20=D0=98=D1=81=D0=BF=D1=80=D0=B0=D0=B2?= =?UTF-8?q?=D0=B8=D0=BB=20=D0=BE=D0=B1=D1=80=D0=B0=D0=B1=D0=BE=D1=82=D0=BA?= =?UTF-8?q?=D1=83=20=D0=BD=D0=B0=D1=87=D0=B0=D0=BB=D0=B0=20=D1=81=D0=BE=20?= =?UTF-8?q?=D1=81=D1=82=D0=B0=D1=82=D1=83=D1=81=D0=BE=D0=BC=20"Oncoming"?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/utils/base/base.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/utils/base/base.py b/src/utils/base/base.py index 88ddea0..49a333f 100644 --- a/src/utils/base/base.py +++ b/src/utils/base/base.py @@ -365,9 +365,11 @@ class BasePointPassportFormer: times:pd.Series, dataframe: pd.DataFrame) -> list[list[float]]: start_idx, finish_idx = self._find_indexes(signal, dataframe) + if start_idx[0] > finish_idx[0]: + start_idx = np.insert(start_idx, 0, 0) + print (start_idx) start_list = times.loc[start_idx].tolist() end_list = times.loc[finish_idx].tolist() - #if start_idx[0] > finish_idx[0]: if len(start_list) - len(end_list) == 1: end_list.append(float(times[len(times)-1])) return start_list, end_list