diff --git a/airflow/models.py b/airflow/models.py index fd1349ecf5860..2b6c5c9e6fac7 100644 --- a/airflow/models.py +++ b/airflow/models.py @@ -619,7 +619,16 @@ def __init__(self, task, execution_date, state=None): if state: self.state = state - def command( + def _key(self): + return (self.task_id, self.dag_id, self.execution_date) + + def __eq__(self, other): + return self._key() == other._key() + + def __hash__(self): + return hash(self._key()) + + def command( self, mark_success=False, ignore_dependencies=False,