add all completed task IDs to Scheduler.all_done

IPython/zmq/parallel/dependency.py

@@ -80,15 +80,11 b' class Dependency(set):'
         if len(self) == 0:
             return True
         if self.mode == 'all':
-            for dep in self:
-                if dep not in completed:
-                    return False
-            return True
+            return self.issubset(completed)
         elif self.mode == 'any':
-            for com in completed:
-                if com in self.dependencies:
-                    return True
-            return False
+            return not self.isdisjoint(completed)
+        else:
+            raise NotImplementedError("Only any|all supported, not %r"%mode)
     
     def as_dict(self):
         """Represent this dependency as a dict. For json compatibility."""

IPython/zmq/parallel/scheduler.py

@@ -30,9 +30,9 b' from IPython.external.decorator import decorator'
 
 @decorator
 def logged(f,self,*args,**kwargs):
-    # print ("#--------------------")
-    # print ("%s(*%s,**%s)"%(f.func_name, args, kwargs))
-    # print ("#--")
+    print ("#--------------------")
+    print ("%s(*%s,**%s)"%(f.func_name, args, kwargs))
+    print ("#--")
     return f(self,*args, **kwargs)
 
 #----------------------------------------------------------------------
@@ -340,6 +340,7 b' class TaskScheduler(object):'
         msg_id = parent['msg_id']
         self.pending[engine].pop(msg_id)
         self.completed[engine].add(msg_id)
+        self.all_done.add(msg_id)
         
         self.update_dependencies(msg_id)