fix: исправлен выбор смещения индекса при формировании временных границ точки

This commit is contained in:
Andrew 2024-12-09 17:48:33 +03:00
parent 4262450df1
commit dcd04e21b6
2 changed files with 1 additions and 3 deletions

View File

@ -1,6 +1,5 @@
from utils.base.base import BasePointPassportFormer, BaseIdealDataBuilder from utils.base.base import BasePointPassportFormer, BaseIdealDataBuilder
import pandas as pd import pandas as pd
from typing import Any
@ -51,7 +50,7 @@ class PassportFormer(BasePointPassportFormer):
if time_is_valid: if time_is_valid:
idx_shift = True if events[self._stages[-1]][1][0] == 0 else False idx_shift = True if events[self._stages[-1]][0][0] == 0 else False
for i in range(point_quantity): for i in range(point_quantity):
operator_settings = { operator_settings = {
@ -69,7 +68,6 @@ class PassportFormer(BasePointPassportFormer):
print(f"Cache miss. Computed and cached.") print(f"Cache miss. Computed and cached.")
idx = i+1 if idx_shift else i idx = i+1 if idx_shift else i
point_timeframe = [events[self._stages[0]][0][i], events[self._stages[-1]][1][idx]] point_timeframe = [events[self._stages[0]][0][i], events[self._stages[-1]][1][idx]]
point_events = {key: [value[0][i], value[1][i]] for key, value in events.items()} point_events = {key: [value[0][i], value[1][i]] for key, value in events.items()}
points_pocket.append([point_timeframe, ideal_data, point_events]) points_pocket.append([point_timeframe, ideal_data, point_events])