diff --git a/cloudquery/sdk/schema/column.py b/cloudquery/sdk/schema/column.py index 4473d57..07486e5 100644 --- a/cloudquery/sdk/schema/column.py +++ b/cloudquery/sdk/schema/column.py @@ -45,15 +45,15 @@ def __eq__(self, __value: object) -> bool: def to_arrow_field(self): metadata = { - arrow.METADATA_PRIMARY_KEY: arrow.METADATA_TRUE - if self.primary_key - else arrow.METADATA_FALSE, - arrow.METADATA_UNIQUE: arrow.METADATA_TRUE - if self.unique - else arrow.METADATA_FALSE, - arrow.METADATA_INCREMENTAL: arrow.METADATA_TRUE - if self.incremental_key - else arrow.METADATA_FALSE, + arrow.METADATA_PRIMARY_KEY: ( + arrow.METADATA_TRUE if self.primary_key else arrow.METADATA_FALSE + ), + arrow.METADATA_UNIQUE: ( + arrow.METADATA_TRUE if self.unique else arrow.METADATA_FALSE + ), + arrow.METADATA_INCREMENTAL: ( + arrow.METADATA_TRUE if self.incremental_key else arrow.METADATA_FALSE + ), } return pa.field(self.name, self.type, metadata=metadata) diff --git a/cloudquery/sdk/schema/table.py b/cloudquery/sdk/schema/table.py index 89adb29..71b3d29 100644 --- a/cloudquery/sdk/schema/table.py +++ b/cloudquery/sdk/schema/table.py @@ -88,9 +88,9 @@ def to_arrow_schema(self): arrow.METADATA_TABLE_DESCRIPTION: self.description, arrow.METADATA_TABLE_TITLE: self.title, arrow.METADATA_TABLE_DEPENDS_ON: self.parent.name if self.parent else "", - arrow.METADATA_INCREMENTAL: arrow.METADATA_TRUE - if self.is_incremental - else arrow.METADATA_FALSE, + arrow.METADATA_INCREMENTAL: ( + arrow.METADATA_TRUE if self.is_incremental else arrow.METADATA_FALSE + ), } for column in self.columns: fields.append(column.to_arrow_field()) diff --git a/setup.py b/setup.py index 22bdb2d..4ce6522 100644 --- a/setup.py +++ b/setup.py @@ -12,7 +12,7 @@ dependencies = [ "cloudquery-plugin-pb==0.0.21", "exceptiongroup==1.2.0", - "black==23.12.1", + "black==24.1.1", "grpcio==1.60.0", "grpcio-tools==1.60.0", "iniconfig==2.0.0",