OrderedDict implemetnation of OrderedSet for Python 2.4

theano/misc/ordered_set.py

-# From http://code.activestate.com/recipes/576694/
-
-KEY, PREV, NEXT = range(3)
-
+MutableSet = None
 try:
     from collections import MutableSet
 except ImportError:
     # Python 2.4
-    from sets import Set as MutableSet
+    pass
+from theano.gof.python25 import OrderedDict
 
+if MutableSet is not None:
+    # From http://code.activestate.com/recipes/576694/
+    KEY, PREV, NEXT = range(3)
 
-class OrderedSet(MutableSet):
+    class OrderedSet(MutableSet):
 
+        # Added by IG-- pre-existing theano code expected sets
+        #   to have this method
         def update(self, container):
             # only allowed ordered containers
             assert isinstance(container, (list, OrderedSet))
@@ -78,6 +81,57 @@ class OrderedSet(MutableSet):
 
         def __del__(self):
             self.clear()                    # remove circular references
+else:
+    # Python 2.4
+    class OrderedSet(object):
+        """
+        An implementation of OrderedSet based on the keys of
+        an OrderedDict.
+        """
+        def __init__(self, iterable=None):
+            self.data = OrderedDict()
+            if iterable is not None:
+                self.update(iterable)
+
+        def update(self, container):
+            # only allowed ordered containers
+            assert isinstance(container, (list, OrderedSet))
+            for elem in container:
+                self.add(elem)
+
+        def add(self, key):
+            self.data[key] = None
+
+        def __len__(self, key):
+            return len(self.data)
+
+        def __contains__(self, key):
+            return key in self.data
+
+        def discard(self, key):
+            if key in self.data:
+                del self.data[key]
+
+        def __iter__(self):
+            return self.data.__iter__()
+
+        def __reversed__(self):
+            return self.data.__reversed__()
+
+        def pop(self, last=True):
+            raise NotImplementedError()
+
+        def __eq__(self, other):
+            return type(self) == type(other) and \
+                    self.data == other.data
+
+        def __del__(self):
+            # Remove circular references
+            self.data.clear()
+
+
+
+
 
 
 if __name__ == '__main__':