code: def clean_db(session):
      yield
      session.query(DagRun).delete()
      session.query(TaskInstance).delete()
  
file_path: airflow/tests/ti_deps/deps/test_not_previously_skipped_dep.py
imports: from __future__ import annotations
  import pendulum
  import pytest
  from airflow.models import DagRun, TaskInstance
  from airflow.operators.empty import EmptyOperator
  from airflow.operators.python import BranchPythonOperator
  from airflow.ti_deps.dep_context import DepContext
  from airflow.ti_deps.deps.not_previously_skipped_dep import NotPreviouslySkippedDep
  from airflow.utils.state import State
  from airflow.utils.types import DagRunType
label_desc: Use image node_id 1 for calling a global function with example usage: clean_db(session) without return types