From 6c29071c9b5ad7ae30bdcde686510d1b48599457 Mon Sep 17 00:00:00 2001 From: Steinthor Palsson Date: Thu, 12 Sep 2024 21:50:22 -0400 Subject: [PATCH] Fix destination setup in test --- tests/load/utils.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/load/utils.py b/tests/load/utils.py index d436172dbe..9ee99eef3d 100644 --- a/tests/load/utils.py +++ b/tests/load/utils.py @@ -220,7 +220,11 @@ def setup_pipeline( """Convenience method to setup pipeline with this configuration""" self.dev_mode = dev_mode - destination = kwargs.pop("destination", None) or self.destination_factory(**kwargs) + destination = kwargs.pop("destination") + if destination is None: + destination = self.destination_factory(**kwargs) + else: + self.setup() pipeline = dlt.pipeline( pipeline_name=pipeline_name, destination=destination,