Update lru_cache.py

[] operator can raise KeyError in dict.
pull/153/head
Anton Hulikau 2018-04-03 01:53:36 +03:00 committed by GitHub
parent ea262de564
commit 4eae44bcd8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -34,7 +34,7 @@ class Cache(object):
Accessing a node updates its position to the front of the LRU list.
"""
node = self.lookup[query]
node = self.lookup.get(query)
if node is None:
return None
self.linked_list.move_to_front(node)
@ -47,7 +47,7 @@ class Cache(object):
If the entry is new and the cache is at capacity, removes the oldest entry
before the new entry is added.
"""
node = self.lookup[query]
node = self.lookup.get(query)
if node is not None:
# Key exists in cache, update the value
node.results = results