Close/Flush byte stream in s3 hook load_string and load_bytes (apache…

…#9211)
MDrakos · Jun 15, 2020 · a80cd25 · a80cd25
1 parent 8b0073d
commit a80cd25
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/airflow/providers/amazon/aws/hooks/s3.py b/airflow/providers/amazon/aws/hooks/s3.py
@@ -514,6 +514,7 @@ def load_string(self,
         bytes_data = string_data.encode(encoding)
         file_obj = io.BytesIO(bytes_data)
         self._upload_file_obj(file_obj, key, bucket_name, replace, encrypt, acl_policy)
+        file_obj.close()
 
     @provide_bucket_name
     @unify_bucket_name_and_key
@@ -548,6 +549,7 @@ def load_bytes(self,
         """
         file_obj = io.BytesIO(bytes_data)
         self._upload_file_obj(file_obj, key, bucket_name, replace, encrypt, acl_policy)
+        file_obj.close()
 
     @provide_bucket_name
     @unify_bucket_name_and_key