From fb61f6804d2e3811045171700d1c0d94726e07a6 Mon Sep 17 00:00:00 2001 From: Leon Luttenberger Date: Wed, 13 Mar 2024 13:51:28 -0500 Subject: [PATCH] test: Add queuing for Neptune bulk load tests --- tests/unit/test_neptune.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/unit/test_neptune.py b/tests/unit/test_neptune.py index 2713750c7..d7236c3fa 100644 --- a/tests/unit/test_neptune.py +++ b/tests/unit/test_neptune.py @@ -236,6 +236,7 @@ def test_gremlin_bulk_load( iam_role=neptune_load_iam_role_arn, use_threads=use_threads, keep_files=keep_files, + queue_request="TRUE", ) res_df = wr.neptune.execute_gremlin(client, f"g.V().hasLabel('{label}').valueMap().with(WithOptions.tokens)") @@ -295,6 +296,7 @@ def test_gremlin_bulk_load_from_files( client=client, path=path, iam_role=neptune_load_iam_role_arn, + queue_request="TRUE", parser_configuration=parser_config, ) res_df = wr.neptune.execute_gremlin(client, f"g.V().hasLabel('{label}').valueMap().with(WithOptions.tokens)")