dag_id = "test_clear_set_dagrun_state"
        self._clean_up(dag_id)
        task_id = "t1"
        dag = DAG(dag_id, start_date=DEFAULT_DATE, max_active_runs=1)
        @dag.task
        def make_arg_lists():
            return [[1], [2], [{"a": "b"}]]
        def consumer(value):
            print(value)
        mapped = PythonOperator.partial(task_id=task_id, dag=dag, python_callable=consumer).expand(
            op_args=make_arg_lists()
        )
        session = settings.Session()
        dagrun_1 = dag.create_dagrun(
            run_type=DagRunType.BACKFILL_JOB,
            state=State.FAILED,
            start_date=DEFAULT_DATE,
            execution_date=DEFAULT_DATE,
            session=session,
        )
        expand_mapped_task(mapped, dagrun_1.run_id, "make_arg_lists", length=2, session=session)
        upstream_ti = dagrun_1.get_task_instance("make_arg_lists", session=session)
        ti = dagrun_1.get_task_instance(task_id, map_index=0, session=session)
        ti2 = dagrun_1.get_task_instance(task_id, map_index=1, session=session)
        upstream_ti.state = State.SUCCESS
        ti.state = State.SUCCESS
        ti2.state = State.SUCCESS
        session.flush()
        dag.clear(
            task_ids=[(task_id, 0), ("make_arg_lists")],