Skip to content

Commit

Permalink
fix: Add missing argument in Build, Test, and Ls operators
Browse files Browse the repository at this point in the history
  • Loading branch information
tomasfarias committed Dec 9, 2021
1 parent 0326956 commit 0009bfd
Showing 1 changed file with 6 additions and 11 deletions.
17 changes: 6 additions & 11 deletions airflow_dbt_python/operators/dbt.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,12 +18,7 @@
from airflow.models.baseoperator import BaseOperator
from airflow.models.xcom import XCOM_RETURN_KEY
from airflow.utils.decorators import apply_defaults
from airflow_dbt_python.hooks.dbt import (
BaseConfig,
IndirectSelection,
LogFormat,
Output,
)
from airflow_dbt_python.hooks.dbt import BaseConfig, LogFormat, Output


class DbtBaseOperator(BaseOperator):
Expand Down Expand Up @@ -394,6 +389,7 @@ def __init__(
select: Optional[list[str]] = None,
exclude: Optional[list[str]] = None,
selector_name: Optional[str] = None,
indirect_selection: Optional[str] = None,
**kwargs,
) -> None:
super().__init__(**kwargs)
Expand All @@ -402,6 +398,7 @@ def __init__(
self.exclude = exclude
self.selector_name = selector_name
self.select = select or models
self.indirect_selection = indirect_selection

@property
def command(self) -> str:
Expand Down Expand Up @@ -563,11 +560,7 @@ def __init__(
Output.from_str(dbt_output) if dbt_output is not None else None
)
self.output_keys = output_keys
self.indirect_selection = (
IndirectSelection.from_str(indirect_selection)
if indirect_selection is not None
else None
)
self.indirect_selection = indirect_selection

@property
def command(self) -> str:
Expand Down Expand Up @@ -675,6 +668,7 @@ def __init__(
singular: Optional[bool] = None,
generic: Optional[bool] = None,
show: Optional[bool] = None,
indirect_selection: Optional[str] = None,
**kwargs,
) -> None:
super().__init__(**kwargs)
Expand All @@ -685,6 +679,7 @@ def __init__(
self.singular = singular
self.generic = generic
self.show = show
self.indirect_selection = indirect_selection

@property
def command(self) -> str:
Expand Down

0 comments on commit 0009bfd

Please sign in to comment.