diff --git a/qlib/data/storage/file_storage.py b/qlib/data/storage/file_storage.py index 8a100a2d19e..e2bc5c3679a 100644 --- a/qlib/data/storage/file_storage.py +++ b/qlib/data/storage/file_storage.py @@ -156,7 +156,7 @@ def clear(self) -> None: def index(self, value: CalVT) -> int: self.check() calendar = self._read_calendar() - return int(np.argwhere(calendar == value)[0]) + return calendar.index(value) def insert(self, index: int, value: CalVT): calendar = self._read_calendar()