TaskFlow API DAGs not recognized

[monometa]

Describe the bug
It seems that the library does not support TaskFlow API DAGs.

To Reproduce

• Python version: 3.10.12
• Python dependencies: Pastebin link

Steps to reproduce the behavior:

1. Copy the TaskFlow API DAG from the official docs: https://airflow.apache.org/docs/apache-airflow/stable/tutorial/taskflow.html#example-taskflow-api-pipeline
2. Save it locally as example_dag.py
3. Run the command: airflow-parse-bench --path example_dag.py
4. Observe the error:

ERROR - No valid DAGs found in example_dag.py

Expected behavior
The library should support Airflow TaskFlow API DAGs and reflect parse time metrics

Screenshots

Desktop (please complete the following information):

• OS: Ubuntu

[AlvaroCavalcante]

Great observation @monometa! Indeed, I haven't tested DAGs with TaskFlow API, I completely forgot about it as I normally don't use it! I'll definitely introduce support for this feature in the future! If you want to contribute to the project, please feel free to open a PR to address this or any other issue that you want!