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