fix bug

airbytehq · Jan 15, 2024 · 6de583a · 6de583a
1 parent ab4bd79
commit 6de583a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/airbyte-ci/connectors/pipelines/pipelines/airbyte_ci/poetry/pipeline.py b/airbyte-ci/connectors/pipelines/pipelines/airbyte_ci/poetry/pipeline.py
@@ -67,7 +67,7 @@ def from_connector_context(connector_context: ConnectorContext) -> "PyPIPublishC
             build_docker_image = connector_context.connector.metadata["connectorBuildOptions"]["baseImage"]
         else:
             build_docker_image = "mwalbeck/python-poetry"
-        return PyPIPublishContext(
+        pypi_context = PyPIPublishContext(
             pypi_token=os.environ["PYPI_TOKEN"],
             test_pypi=True,  # TODO: Go live
             package_path=str(connector_context.connector.code_directory),
@@ -85,6 +85,7 @@ def from_connector_context(connector_context: ConnectorContext) -> "PyPIPublishC
             ci_context=connector_context.ci_context,
             ci_gcs_credentials=connector_context.ci_gcs_credentials,
         )
+        pypi_context.dagger_client = connector_context.dagger_client
 
 
 class PublishToPyPI(Step):