Skip to content

Commit b1c64ce

Browse files
committed
Update _filter implementation
1 parent cc46256 commit b1c64ce

File tree

1 file changed

+11
-8
lines changed

1 file changed

+11
-8
lines changed

airbyte_cdk/sources/declarative/schema/dynamic_schema_loader.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -174,17 +174,20 @@ def _transform(
174174
return properties
175175

176176
def _filter(
177-
self,
178-
properties: Mapping[str, Any],
177+
self,
178+
properties: Mapping[str, Any],
179179
) -> Mapping[str, Any]:
180-
if self.schema_filter:
181-
filtered_properties = {}
182-
for property in self.schema_filter.filter_records(properties.items(), {}):
183-
filtered_properties[property[0]] = property[1]
184-
return filtered_properties
185-
else:
180+
if not self.schema_filter:
186181
return properties
187182

183+
filtered_properties = {}
184+
for item in self.schema_filter.filter_records(
185+
({k: v} for k, v in properties.items()),
186+
{},
187+
):
188+
filtered_properties.update(item)
189+
return filtered_properties
190+
188191
def _get_key(
189192
self,
190193
raw_schema: MutableMapping[str, Any],

0 commit comments

Comments
 (0)