diff --git a/qlib/data/storage/file_storage.py b/qlib/data/storage/file_storage.py index 8a100a2d1..e2bc5c367 100644 --- a/qlib/data/storage/file_storage.py +++ b/qlib/data/storage/file_storage.py @@ -156,7 +156,7 @@ class FileCalendarStorage(FileStorageMixin, CalendarStorage): 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()