Fix in-place trait set operations.

enthought · Mar 16, 2016 · eab0984 · eab0984
1 parent 2ee132b
commit eab0984
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/traits/trait_handlers.py b/traits/trait_handlers.py
@@ -2757,6 +2757,7 @@ def update ( self, value ):
         if not hasattr(self, 'trait'):
             return set.update(self, value)
         try:
+            value = set(value)
             added = value.difference( self )
             if len( added ) > 0:
                 object   = self.object()
@@ -2796,6 +2797,7 @@ def difference_update ( self, value ):
     def symmetric_difference_update ( self, value ):
         if not hasattr(self, 'trait'):
             return set.symmetric_difference_update(self, value)
+        value = set(value)
         removed = self.intersection( value )
         added   = value.difference( self )
         if (len( removed ) > 0) or (len( added ) > 0):