liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-18 01:44:34 +08:00
Code Activity

fix IndexError of the last trading day in backtest calendar (#751)

Browse Source
This commit is contained in:
Pengrong Zhu
2021-12-17 11:11:56 +08:00
committed by GitHub
parent 6c83632fc4
commit 957f9a18e9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
qlib/backtest/utils.py
View File

@@ -55,9 +55,9 @@ class TradeCalendarManager:
self.start_time = pd.Timestamp(start_time) if start_time else None self.start_time = pd.Timestamp(start_time) if start_time else None
self.end_time = pd.Timestamp(end_time) if end_time else None self.end_time = pd.Timestamp(end_time) if end_time else None
_calendar = Cal.calendar(freq=freq) _calendar = Cal.calendar(freq=freq, future=True)
self._calendar = _calendar self._calendar = _calendar
_, _, _start_index, _end_index = Cal.locate_index(start_time, end_time, freq=freq) _, _, _start_index, _end_index = Cal.locate_index(start_time, end_time, freq=freq, future=True)
self.start_index = _start_index self.start_index = _start_index
self.end_index = _end_index self.end_index = _end_index
self.trade_len = _end_index - _start_index + 1 self.trade_len = _end_index - _start_index + 1