Parcourir la source

Fix dict KeyError (#153)

Anton Hulikau il y a 7 ans
Parent
commit
a70a8f3a04
1 fichiers modifiés avec 2 ajouts et 2 suppressions
  1. 2 2
      solutions/object_oriented_design/lru_cache/lru_cache.py

+ 2 - 2
solutions/object_oriented_design/lru_cache/lru_cache.py

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