1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-04 19:41:00 +08:00

make code cleaner

This commit is contained in:
hadrianl
2020-12-22 18:38:59 +08:00
committed by you-n-g
parent 40bbafcaab
commit 5efe82fb56

View File

@@ -174,16 +174,16 @@ class MemCache:
size_limit = C.mem_cache_size_limit if mem_cache_size_limit is None else mem_cache_size_limit
if limit_type == "length":
self.__calendar_mem_cache = MemCacheLengthUnit(size_limit)
self.__instrument_mem_cache = MemCacheLengthUnit(size_limit)
self.__feature_mem_cache = MemCacheLengthUnit(size_limit)
klass = MemCacheLengthUnit
elif limit_type == "sizeof":
self.__calendar_mem_cache = MemCacheSizeofUnit(size_limit)
self.__instrument_mem_cache = MemCacheSizeofUnit(size_limit)
self.__feature_mem_cache = MemCacheSizeofUnit(size_limit)
klass = MemCacheSizeofUnit
else:
raise ValueError(f"limit_type must be length or sizeof, your limit_type is {limit_type}")
self.__calendar_mem_cache = klass(size_limit)
self.__instrument_mem_cache = klass(size_limit)
self.__feature_mem_cache = klass(size_limit)
def __getitem__(self, key):
if key == "c":
return self.__calendar_mem_cache