Skip to content

Commit 39d5103

Browse files
authored
fix(provider): Fix kwargs handling in Azure Data Lake Storage V2 Hook methods (#51847)
1 parent 7ff2b7a commit 39d5103

File tree

1 file changed

+4
-4
lines changed
  • providers/microsoft/azure/src/airflow/providers/microsoft/azure/hooks

1 file changed

+4
-4
lines changed

providers/microsoft/azure/src/airflow/providers/microsoft/azure/hooks/data_lake.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -424,7 +424,7 @@ def create_directory(
424424
:param file_system_name: Name of the file system or instance of FileSystemProperties.
425425
:param directory_name: Name of the directory which needs to be created in the file system.
426426
"""
427-
result = self.get_file_system(file_system_name).create_directory(directory_name, kwargs)
427+
result = self.get_file_system(file_system_name).create_directory(directory_name, **kwargs)
428428
return result
429429

430430
def get_directory_client(
@@ -479,7 +479,7 @@ def upload_file(
479479
"""
480480
file_client = self.create_file(file_system_name, file_name)
481481
with open(file_path, "rb") as data:
482-
file_client.upload_data(data, overwrite=overwrite, kwargs=kwargs)
482+
file_client.upload_data(data, overwrite=overwrite, **kwargs)
483483

484484
def upload_file_to_directory(
485485
self,
@@ -500,9 +500,9 @@ def upload_file_to_directory(
500500
:param overwrite: Boolean flag to overwrite an existing file or not.
501501
"""
502502
directory_client = self.get_directory_client(file_system_name, directory_name=directory_name)
503-
file_client = directory_client.create_file(file_name, kwargs=kwargs)
503+
file_client = directory_client.create_file(file_name, **kwargs)
504504
with open(file_path, "rb") as data:
505-
file_client.upload_data(data, overwrite=overwrite, kwargs=kwargs)
505+
file_client.upload_data(data, overwrite=overwrite, **kwargs)
506506

507507
def list_files_directory(
508508
self, file_system_name: FileSystemProperties | str, directory_name: str

0 commit comments

Comments
 (0)