From 5efe82fb566d640040381177a69c578ebf0f5323 Mon Sep 17 00:00:00 2001
From: hadrianl <137150224@qq.com>
Date: Tue, 22 Dec 2020 18:38:59 +0800
Subject: [PATCH] make code cleaner

---
 qlib/data/cache.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/qlib/data/cache.py b/qlib/data/cache.py
index 00cbcec9e..0f753b7b5 100644
--- a/qlib/data/cache.py
+++ b/qlib/data/cache.py
@@ -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