Skip to content

Commit

Permalink
Merge pull request #20 from MeltanoLabs/bugfix-pks-not-populating
Browse files Browse the repository at this point in the history
fix: PKs not populating
  • Loading branch information
pnadolny13 authored Apr 18, 2023
2 parents 442a431 + 4cd2a55 commit be786d3
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions tap_dynamodb/streams.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,11 +34,7 @@ def __init__(
self._table_name: str = name
self._schema: dict = {}
self._infer_schema_sample_size = infer_schema_sample_size
super().__init__(
tap=tap,
schema=self.schema,
name=name,
)
super().__init__(name=name, tap=tap)

def get_records(self, context: dict | None) -> Iterable[dict]:
for batch in self._dynamodb_conn.get_items_iter(self._table_name):
Expand All @@ -59,7 +55,7 @@ def schema(self) -> dict:
self._table_name,
self._infer_schema_sample_size,
)
self.primary_keys = self._dynamodb_conn.get_table_key_properties(
self._primary_keys = self._dynamodb_conn.get_table_key_properties(
self._table_name
)
return self._schema

0 comments on commit be786d3

Please sign in to comment.