Skip to content

Commit

Permalink
Add kwargs param in DocsOperator method upload_to_cloud_storage
Browse files Browse the repository at this point in the history
  • Loading branch information
pankajastro committed Dec 23, 2024
1 parent dada5cf commit 2d7326c
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions cosmos/operators/local.py
Original file line number Diff line number Diff line change
Expand Up @@ -862,7 +862,7 @@ def __init__(
self.callback = self.upload_to_cloud_storage

@abstractmethod
def upload_to_cloud_storage(self, project_dir: str) -> None:
def upload_to_cloud_storage(self, project_dir: str, **kwargs: Any) -> None:
"""Abstract method to upload the generated documentation to cloud storage."""


Expand Down Expand Up @@ -893,7 +893,7 @@ def __init__(
kwargs["connection_id"] = aws_conn_id
super().__init__(*args, **kwargs)

def upload_to_cloud_storage(self, project_dir: str) -> None:
def upload_to_cloud_storage(self, project_dir: str, **kwargs: Any) -> None:
"""Uploads the generated documentation to S3."""
self.log.info(
'Attempting to upload generated docs to S3 using S3Hook("%s")',
Expand Down Expand Up @@ -959,7 +959,7 @@ def __init__(
kwargs["bucket_name"] = container_name
super().__init__(*args, **kwargs)

def upload_to_cloud_storage(self, project_dir: str) -> None:
def upload_to_cloud_storage(self, project_dir: str, **kwargs: Any) -> None:
"""Uploads the generated documentation to Azure Blob Storage."""
self.log.info(
'Attempting to upload generated docs to Azure Blob Storage using WasbHook(conn_id="%s")',
Expand Down Expand Up @@ -1003,7 +1003,7 @@ class DbtDocsGCSLocalOperator(DbtDocsCloudLocalOperator):

ui_color = "#4772d5"

def upload_to_cloud_storage(self, project_dir: str) -> None:
def upload_to_cloud_storage(self, project_dir: str, **kwargs: Any) -> None:
"""Uploads the generated documentation to Google Cloud Storage"""
self.log.info(
'Attempting to upload generated docs to Storage using GCSHook(conn_id="%s")',
Expand Down

0 comments on commit 2d7326c

Please sign in to comment.