diff --git a/inference/core/cache/lru_cache.py b/inference/core/cache/lru_cache.py index 5f801f3451..67ffc3bf04 100644 --- a/inference/core/cache/lru_cache.py +++ b/inference/core/cache/lru_cache.py @@ -17,11 +17,10 @@ def enforce_size(self): self.cache.popitem(last=False) def get(self, key): - try: - value = self.cache.pop(key) - self.cache[key] = value - return value - except KeyError: + if key in self.cache: + self.cache.move_to_end(key) + return self.cache[key] + else: return None def set(self, key, value):