diff --git a/pyproject.toml b/pyproject.toml index 49ba8835..86c889d5 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -67,4 +67,7 @@ push = false [tool.bumpver.file_patterns] "pyproject.toml" = ['current_version = "{version}"', 'version = "{version}"'] -"src/msgraph_core/_constants.py" = ["{version}"] \ No newline at end of file +"src/msgraph_core/_constants.py" = ["{version}"] + +[tool.poetry.packages] +include = ["src/msgraph_core"] \ No newline at end of file diff --git a/src/msgraph_core/models/upload_result.py b/src/msgraph_core/models/upload_result.py index a8edf258..1adc8af4 100644 --- a/src/msgraph_core/models/upload_result.py +++ b/src/msgraph_core/models/upload_result.py @@ -1,4 +1,4 @@ -from typing import Any, Callable, Dict, List, Optional, TypeVar +from typing import Any, Callable, Dict, List, Optional, TypeVar, Generic from dataclasses import dataclass from datetime import datetime @@ -50,9 +50,9 @@ def serialize(self, writer: SerializationWriter) -> None: writer.write_additional_data_value(self.additional_data) -class UploadResult: +class UploadResult(Generic[T]): - def __init__(self): + def __init__(self) -> None: self.upload_session: Optional[UploadSessionDataHolder] = None self.item_response: Optional[T] = None self.location: Optional[str] = None diff --git a/src/msgraph_core/py.typed.txt b/src/msgraph_core/py.typed.txt new file mode 100644 index 00000000..e69de29b diff --git a/src/msgraph_core/tasks/large_file_upload.py b/src/msgraph_core/tasks/large_file_upload.py index dcb5c5c0..88b9422b 100644 --- a/src/msgraph_core/tasks/large_file_upload.py +++ b/src/msgraph_core/tasks/large_file_upload.py @@ -133,7 +133,7 @@ async def upload(self, after_chunk_upload: Optional[Callable] = None): logging.error("Error uploading chunk %s", error) finally: self.chunks -= 1 - upload_result = UploadResult() + upload_result: UploadResult[Any] = UploadResult() upload_result.item_response = response upload_result.location = self.upload_session.upload_url return upload_result