diff --git a/src/controller/__pycache__/passportFormer.cpython-310.pyc b/src/controller/__pycache__/passportFormer.cpython-310.pyc index 95d7b18..a259d12 100644 Binary files a/src/controller/__pycache__/passportFormer.cpython-310.pyc and b/src/controller/__pycache__/passportFormer.cpython-310.pyc differ diff --git a/src/controller/passportFormer.py b/src/controller/passportFormer.py index 8abb3ba..77d90ed 100644 --- a/src/controller/passportFormer.py +++ b/src/controller/passportFormer.py @@ -1,6 +1,5 @@ from utils.base.base import BasePointPassportFormer, BaseIdealDataBuilder import pandas as pd -from typing import Any @@ -51,7 +50,7 @@ class PassportFormer(BasePointPassportFormer): 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): operator_settings = { @@ -69,7 +68,6 @@ class PassportFormer(BasePointPassportFormer): print(f"Cache miss. Computed and cached.") idx = i+1 if idx_shift else i 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()} points_pocket.append([point_timeframe, ideal_data, point_events])