def METHOD_NAME(model_name: str, include_upstream: bool, include_downstream: bool) -> None:
    result.add(model_name)
    if include_upstream:
        result.update(self._dag.upstream(model_name))
    if include_downstream:
        result.update(self._dag.downstream(model_name))