Skip to content

Commit 395beed

Browse files
author
dakodakov
authored
vdk-test-utils: adopt ingester changes (#3119)
Adopt ingester changes in test utils. Ref: 03dde45 Signed-off-by: Dako Dakov <[email protected]>
1 parent 28ed822 commit 395beed

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

projects/vdk-plugins/vdk-test-utils/src/vdk/plugin/test_utils/ingest_util_plugins.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ def pre_ingest_process(
2626
target: Optional[str] = None,
2727
collection_id: Optional[str] = None,
2828
metadata: IngestionMetadata = None,
29+
method: Optional[str] = None,
2930
) -> Tuple[List[Dict], Optional[IngestionMetadata]]:
3031
processed_payload = [{k: str(v) for (k, v) in i.items()} for i in payload]
3132
return processed_payload, metadata
@@ -56,6 +57,7 @@ def pre_ingest_process(
5657
destination_table: Optional[str] = None,
5758
target: Optional[str] = None,
5859
collection_id: Optional[str] = None,
60+
method: Optional[str] = None,
5961
) -> Tuple[List[Dict], Optional[IngestionMetadata]]:
6062
processed_payloads = []
6163
start_time = datetime.datetime(year=2022, month=1, day=27, hour=16)
@@ -91,6 +93,7 @@ def post_ingest_process(
9193
destination_table: Optional[str] = None,
9294
target: Optional[str] = None,
9395
collection_id: Optional[str] = None,
96+
method: Optional[str] = None,
9497
) -> Optional[IngestionMetadata]:
9598
start_time = None
9699
if metadata:

0 commit comments

Comments
 (0)