diff --git a/core/dbt/artifacts/resources/v1/config.py b/core/dbt/artifacts/resources/v1/config.py index 79529aa9e99..b053114b9ea 100644 --- a/core/dbt/artifacts/resources/v1/config.py +++ b/core/dbt/artifacts/resources/v1/config.py @@ -8,7 +8,7 @@ from dbt import hooks from dbt.artifacts.resources.base import Docs -from dbt.artifacts.resources.types import ModelHookType +from dbt.artifacts.resources.types import BatchSize, ModelHookType from dbt.artifacts.utils.validation import validate_color from dbt_common.contracts.config.base import BaseConfig, CompareBehavior, MergeBehavior from dbt_common.contracts.config.materialization import OnConfigurationChangeOption @@ -81,8 +81,8 @@ class NodeConfig(NodeAndTestConfig): # 'mergebehavior' dictionary materialized: str = "view" incremental_strategy: Optional[str] = None - batch_size: Any = None - lookback: Any = 0 + batch_size: Union[BatchSize, Any] = None + lookback: Union[int, Any] = 0 begin: Union[datetime, Any] = None persist_docs: Dict[str, Any] = field(default_factory=dict) post_hook: List[Hook] = field( @@ -126,7 +126,7 @@ class NodeConfig(NodeAndTestConfig): default_factory=ContractConfig, metadata=MergeBehavior.Update.meta(), ) - event_time: Any = None + event_time: Union[str, Any] = None def __post_init__(self): # we validate that node_color has a suitable value to prevent dbt-docs from crashing