Change the default chunk_size to a clear representation & add docum…

…entation (#22222)
apache · Mar 13, 2022 · c108f26 · c108f26
1 parent af9d85c
commit c108f26
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/airflow/providers/google/suite/hooks/drive.py b/airflow/providers/google/suite/hooks/drive.py
@@ -219,12 +219,13 @@ def upload_file(
  self.log.info("File %s uploaded to gdrive://%s.", local_location, remote_location)
  return file.get("id")
 
- def download_file(self, file_id: str, file_handle: IO, chunk_size: int = 104857600):
+ def download_file(self, file_id: str, file_handle: IO, chunk_size: int = 100 * 1024 * 1024):
  """
  Download a file from Google Drive.
 
  :param file_id: the id of the file
  :param file_handle: file handle used to write the content to
+ :param chunk_size: File will be downloaded in chunks of this many bytes.
  """
  request = self.get_media_request(file_id=file_id)
  self.download_content_from_request(file_handle=file_handle, request=request, chunk_size=chunk_size)