From b1c6f46825e41e675e78b89583fd1d20318f9d4b Mon Sep 17 00:00:00 2001 From: omrdyngc Date: Tue, 17 Jun 2025 10:55:26 +0000 Subject: [PATCH] fix(provider): Fix kwargs handling in Azure Data Lake Storage V2 Hook methods --- .../airflow/providers/microsoft/azure/hooks/data_lake.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/providers/microsoft/azure/src/airflow/providers/microsoft/azure/hooks/data_lake.py b/providers/microsoft/azure/src/airflow/providers/microsoft/azure/hooks/data_lake.py index 3e479c530e5e1..f18ba472015d1 100644 --- a/providers/microsoft/azure/src/airflow/providers/microsoft/azure/hooks/data_lake.py +++ b/providers/microsoft/azure/src/airflow/providers/microsoft/azure/hooks/data_lake.py @@ -424,7 +424,7 @@ def create_directory( :param file_system_name: Name of the file system or instance of FileSystemProperties. :param directory_name: Name of the directory which needs to be created in the file system. """ - result = self.get_file_system(file_system_name).create_directory(directory_name, kwargs) + result = self.get_file_system(file_system_name).create_directory(directory_name, **kwargs) return result def get_directory_client( @@ -479,7 +479,7 @@ def upload_file( """ file_client = self.create_file(file_system_name, file_name) with open(file_path, "rb") as data: - file_client.upload_data(data, overwrite=overwrite, kwargs=kwargs) + file_client.upload_data(data, overwrite=overwrite, **kwargs) def upload_file_to_directory( self, @@ -500,9 +500,9 @@ def upload_file_to_directory( :param overwrite: Boolean flag to overwrite an existing file or not. """ directory_client = self.get_directory_client(file_system_name, directory_name=directory_name) - file_client = directory_client.create_file(file_name, kwargs=kwargs) + file_client = directory_client.create_file(file_name, **kwargs) with open(file_path, "rb") as data: - file_client.upload_data(data, overwrite=overwrite, kwargs=kwargs) + file_client.upload_data(data, overwrite=overwrite, **kwargs) def list_files_directory( self, file_system_name: FileSystemProperties | str, directory_name: str