diff --git a/heat/engine/dependencies.py b/heat/engine/dependencies.py index d488b5a39..f7fb7cafe 100644 --- a/heat/engine/dependencies.py +++ b/heat/engine/dependencies.py @@ -71,6 +71,10 @@ class Node(object): '''Return True if this node is not a leaf (it requires other nodes).''' return bool(self.require) + def __bool__(self): + '''Return True if this node is not a leaf (it requires other nodes).''' + return self.__nonzero__() + def stem(self): '''Return True if this node is a stem (required by nothing).''' return not bool(self.satisfy) diff --git a/heat/engine/scheduler.py b/heat/engine/scheduler.py index ed79448d9..f9bb0f657 100644 --- a/heat/engine/scheduler.py +++ b/heat/engine/scheduler.py @@ -283,6 +283,10 @@ class TaskRunner(object): """Return True if there are steps remaining.""" return not self.done() + def __bool__(self): + """Return True if there are steps remaining.""" + return self.__nonzero__() + def wrappertask(task): """