clean up code

Author: shemogumbe

src/msgraph_core/tasks/large_file_upload.py

@@ -1,10 +1,11 @@
 import os
-from typing import Callable, Optional, List, Tuple, Any, Dict
+from typing import Callable, Optional, List, Tuple, Any, Dict, Union
 from io import BytesIO
 from asyncio import Future
 from datetime import datetime, timedelta, timezone
 import logging
 import httpx
+from httpx import Response
 
 from kiota_abstractions.serialization.parsable import Parsable
 from kiota_abstractions.method import Method
@@ -152,7 +153,10 @@ def next_range(self):
     def next_range(self, value: Optional[str]) -> None:
         self._next_range = value
 
-    async def next_chunk(self, file: BytesIO, range_start: int = 0, range_end: int = 0) -> Future:
+    async def next_chunk(self,
+                         file: BytesIO,
+                         range_start: int = 0,
+                         range_end: int = 0) -> Union[Future, Response]:
         upload_url = self.get_validated_upload_url(self.upload_session)
         if not upload_url:
             raise ValueError('The upload session URL must not be empty.')
@@ -200,7 +204,7 @@ async def last_chunk(
         range_start: int = 0,
         range_end: int = 0,
         parsable_factory: Optional[ParsableFactory] = None
-    ) -> Future:
+    ) -> Union[Future, Response]:
         upload_url = self.get_validated_upload_url(self.upload_session)
         if not upload_url:
             raise ValueError('The upload session URL must not be empty.')