Fix DbtRunLocalOperator.partial() support

[tatiana]

Since Cosmos 1.9.0, users who attempted to use:

DbtRunLocalOperator.partial(task_id="foo", project_dir="foo")

Started facing the issue:

File /usr/local/lib/python3.11/site-packages/airflow/models/baseoperator.py:284, in partial(operator_class, task_id, dag, task_group, start_date, end_date, owner, email, params, resources, trigger_rule, depends_on_past, ignore_first_depends_on_past, wait_for_past_depends_before_skipping, wait_for_downstream, retries, queue, pool, pool_slots, execution_timeout, max_retry_delay, retry_delay, retry_exponential_backoff, priority_weight, weight_rule, sla, map_index_template, max_active_tis_per_dag, max_active_tis_per_dagrun, on_execute_callback, on_failure_callback, on_success_callback, on_retry_callback, on_skipped_callback, run_as_user, executor, executor_config, inlets, outlets, doc, doc_md, doc_json, doc_yaml, doc_rst, task_display_name, logger_name, allow_nested_operators, **kwargs)
281 from airflow.models.dag import DagContext
282 from airflow.utils.task_group import TaskGroupContext
--> 284 validate_mapping_kwargs(operator_class, "partial", kwargs)
286 dag = dag or DagContext.get_current_dag()
287 if dag:

File /usr/local/lib/python3.11/site-packages/airflow/models/mappedoperator.py:123, in validate_mapping_kwargs(op, func, value)
121 names = ", ".join(repr(n) for n in unknown_args)
122 error = f"unexpected keyword arguments {names}"
--> 123 raise TypeError(f"{op.name}.{func}() got {error}")

TypeError: DbtRunLocalOperator.partial() got an unexpected keyword argument 'project_dir'`

This was introduced given the changes in how Cosmos operators subclass, that was introduced to allow to dynamically chose which Airflow operator is run during DAG rendering time.

Closes: #1546

To validate it, we introduced a new small dbt project and an example DAG, and it can be tested by running:

airflow dags test example_task_mapping

Co-authored-by: Ash Berlin-Taylor ash@astronomer.io

[netlify]

✅ Deploy Preview for sunny-pastelito-5ecb04 canceled.

Name	Link
🔨 Latest commit	c2e31ad
🔍 Latest deploy log	https://app.netlify.com/sites/sunny-pastelito-5ecb04/deploys/67d304708975ad0008438913

[cloudflare-workers-and-pages]

Deploying astronomer-cosmos with    Cloudflare Pages

Latest commit:	c2e31ad
Status:	✅  Deploy successful!
Preview URL:	https://cb8d04fa.astronomer-cosmos.pages.dev
Branch Preview URL:	https://issue-1546.astronomer-cosmos.pages.dev

View logs

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 97.43%. Comparing base (d494dcd) to head (c2e31ad).
Report is 1 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #1609   +/-   ##
=======================================
  Coverage   97.43%   97.43%           
=======================================
  Files          80       80           
  Lines        4943     4948    +5     
=======================================
+ Hits         4816     4821    +5     
  Misses        127      127           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[pankajkoti]

I am merging the PR so that we can cherry-pick it in #1607