1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-15 00:36:55 +08:00

Add Cache to avoid frequently loading calendar (#766)

This commit is contained in:
you-n-g
2021-12-23 09:08:52 +08:00
committed by GitHub
parent 350fbe91c9
commit d087054a59
6 changed files with 101 additions and 52 deletions

View File

@@ -10,6 +10,7 @@ Two modes are supported
- server
"""
from __future__ import annotations
import os
import re
@@ -18,7 +19,11 @@ import logging
import platform
import multiprocessing
from pathlib import Path
from typing import Union
from typing import Optional, Union
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from qlib.utils.time import Freq
class Config:
@@ -296,7 +301,9 @@ class QlibConfig(Config):
else:
return QlibConfig.LOCAL_URI
def get_data_uri(self, freq: str = None) -> Path:
def get_data_uri(self, freq: Optional[Union[str, Freq]] = None) -> Path:
if freq is not None:
freq = str(freq) # converting Freq to string
if freq is None or freq not in self.provider_uri:
freq = QlibConfig.DEFAULT_FREQ
_provider_uri = self.provider_uri[freq]