diff --git a/src/dbt_bouncer/dbt_cloud/manifest_latest.py b/src/dbt_bouncer/dbt_cloud/manifest_latest.py index bb7a604b..676006c0 100644 --- a/src/dbt_bouncer/dbt_cloud/manifest_latest.py +++ b/src/dbt_bouncer/dbt_cloud/manifest_latest.py @@ -13,11 +13,20 @@ from pydantic import ConfigDict, Field, constr +class CustomGranularity(BaseParserModel): + model_config = ConfigDict( + extra='forbid', + ) + name: str + column_name: Optional[str] = None + + class TimeSpine(BaseParserModel): model_config = ConfigDict( - extra="forbid", + extra='forbid', ) standard_granularity_column: str + custom_granularities: Optional[List[CustomGranularity]] = None class Metadata(BaseParserModel):