From e7cc889d04febe8144c2530af592637c11141a78 Mon Sep 17 00:00:00 2001 From: Robin Mackaij Date: Tue, 11 Mar 2025 09:19:26 +0000 Subject: [PATCH 1/2] Added py.typed to support mypy when subclassing --- src/DataDriver/py.typed | 0 1 file changed, 0 insertions(+), 0 deletions(-) create mode 100644 src/DataDriver/py.typed diff --git a/src/DataDriver/py.typed b/src/DataDriver/py.typed new file mode 100644 index 0000000..e69de29 From 11d4427c7321734f8568fcdd692af64e9fe797cc Mon Sep 17 00:00:00 2001 From: Robin Mackaij Date: Tue, 11 Mar 2025 09:19:56 +0000 Subject: [PATCH 2/2] Correct annotation for the reader_class argument --- src/DataDriver/DataDriver.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/DataDriver/DataDriver.py b/src/DataDriver/DataDriver.py index 75f8424..d4fad8e 100644 --- a/src/DataDriver/DataDriver.py +++ b/src/DataDriver/DataDriver.py @@ -18,7 +18,7 @@ import traceback from glob import glob from pathlib import Path -from typing import Any, Dict, List, Optional, Union # type: ignore +from typing import Any, Dict, List, Optional, Type, Union # type: ignore from robot.api.logger import console # type: ignore from robot.libraries.BuiltIn import BuiltIn # type: ignore @@ -1357,7 +1357,7 @@ def __init__( lineterminator: str = "\r\n", *, sheet_name: Union[str, int] = 0, - reader_class: Optional[Union[AbstractReaderClass, str]] = None, + reader_class: Optional[Union[Type[AbstractReaderClass], str]] = None, file_search_strategy: str = "PATH", file_regex: str = r"(?i)(.*?)(\.csv)", include: Optional[str] = None,