1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-16 01:06:56 +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 size_limit = C.mem_cache_size_limit if mem_cache_size_limit is None else mem_cache_size_limit
if limit_type == "length": if limit_type == "length":
self.__calendar_mem_cache = MemCacheLengthUnit(size_limit) klass = MemCacheLengthUnit
self.__instrument_mem_cache = MemCacheLengthUnit(size_limit)
self.__feature_mem_cache = MemCacheLengthUnit(size_limit)
elif limit_type == "sizeof": elif limit_type == "sizeof":
self.__calendar_mem_cache = MemCacheSizeofUnit(size_limit) klass = MemCacheSizeofUnit
self.__instrument_mem_cache = MemCacheSizeofUnit(size_limit)
self.__feature_mem_cache = MemCacheSizeofUnit(size_limit)
else: else:
raise ValueError(f"limit_type must be length or sizeof, your limit_type is {limit_type}") 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): def __getitem__(self, key):
if key == "c": if key == "c":
return self.__calendar_mem_cache return self.__calendar_mem_cache