From 31705913c04ed27f846b453085153291732613b4 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Thu, 19 Aug 2021 18:34:19 +0800 Subject: [PATCH] fix(ci): set MGE_FASTRUN_CACHE_TYPE=FILE in ci env GitOrigin-RevId: c4a549480e2e44cd51a48972fb38ff8f31f237ac --- imperative/python/megengine/utils/persistent_cache.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/imperative/python/megengine/utils/persistent_cache.py b/imperative/python/megengine/utils/persistent_cache.py index b0bd351d4..b88e58d29 100644 --- a/imperative/python/megengine/utils/persistent_cache.py +++ b/imperative/python/megengine/utils/persistent_cache.py @@ -19,6 +19,8 @@ from ..version import __version__ class _FakeRedisConn: + cache_file = None + def __init__(self): try: from ..hub.hub import _get_megengine_home @@ -27,8 +29,8 @@ class _FakeRedisConn: os.path.join(_get_megengine_home(), "persistent_cache") ) os.makedirs(cache_dir, exist_ok=True) - cache_file = os.path.join(cache_dir, "cache") - self._dict = shelve.open(cache_file) + self.cache_file = os.path.join(cache_dir, "cache") + self._dict = shelve.open(self.cache_file) self._is_shelve = True except: self._dict = {} -- GitLab