devve1 commited on
Commit
600cdc6
1 Parent(s): 19aff28

Update ordered_multimap_weakref.py

Browse files
Files changed (1) hide show
  1. ordered_multimap_weakref.py +3 -3
ordered_multimap_weakref.py CHANGED
@@ -1,14 +1,14 @@
1
- import weakref
2
  from collections import defaultdict
3
 
4
  class OrderedMultimapWeakRef:
5
  def __init__(self):
6
  self.data = []
7
- self.index_map = weakref.WeakValueDictionary()
8
 
9
  def insert(self, text, title):
10
  self.data.append(text)
11
- self.index_map[len(self.data)-1] = title
12
 
13
  def get(self, key):
14
  if key:
 
1
+ from weak_list import WeakList
2
  from collections import defaultdict
3
 
4
  class OrderedMultimapWeakRef:
5
  def __init__(self):
6
  self.data = []
7
+ self.index_list = WeakList()
8
 
9
  def insert(self, text, title):
10
  self.data.append(text)
11
+ self.index_list.append(title)
12
 
13
  def get(self, key):
14
  if key: