diff --git a/CHANGELOG.rst b/CHANGELOG.rst index 168d8fd0b..3cff1bbb3 100644 --- a/CHANGELOG.rst +++ b/CHANGELOG.rst @@ -1,6 +1,13 @@ Changelog ========= +1.8.1a1 (2024-12-23) +-------------------- + +Bug Fixes + +* Add ``kwargs`` param in DocsOperator method ``upload_to_cloud_storage`` by @pankajastro in #1422 + 1.8.0 (2024-12-20) -------------------- diff --git a/cosmos/__init__.py b/cosmos/__init__.py index 81da409ab..9577b9dbd 100644 --- a/cosmos/__init__.py +++ b/cosmos/__init__.py @@ -6,7 +6,7 @@ Contains dags, task groups, and operators. """ -__version__ = "1.8.0" +__version__ = "1.8.1a1" from cosmos.airflow.dag import DbtDag diff --git a/cosmos/operators/local.py b/cosmos/operators/local.py index dee00114c..c5d47dd48 100644 --- a/cosmos/operators/local.py +++ b/cosmos/operators/local.py @@ -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.""" @@ -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")', @@ -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")', @@ -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")',