|
5 | 5 | import logging
|
6 | 6 | import numbers
|
7 | 7 | import os
|
8 |
| -from typing import Any, Callable, Dict, List, Optional, Tuple, Union |
| 8 | +from typing import Any, Callable, Dict, Optional, Sequence, Union |
9 | 9 |
|
10 | 10 | from ..shared import constants
|
11 | 11 | from ..shared.functions import resolve_env_var_choice, resolve_truthy_env_var_choice
|
@@ -53,7 +53,7 @@ class Tracer:
|
53 | 53 | disabled: bool
|
54 | 54 | Flag to explicitly disable tracing, useful when running/testing locally
|
55 | 55 | `Env POWERTOOLS_TRACE_DISABLED="true"`
|
56 |
| - patch_modules: Tuple[str] |
| 56 | + patch_modules: Optional[Sequence[str]] |
57 | 57 | Tuple of modules supported by tracing provider to patch, by default all modules are patched
|
58 | 58 | provider: BaseProvider
|
59 | 59 | Tracing provider, by default it is aws_xray_sdk.core.xray_recorder
|
@@ -149,7 +149,7 @@ def __init__(
|
149 | 149 | service: Optional[str] = None,
|
150 | 150 | disabled: Optional[bool] = None,
|
151 | 151 | auto_patch: Optional[bool] = None,
|
152 |
| - patch_modules: Optional[Tuple[str]] = None, |
| 152 | + patch_modules: Optional[Sequence[str]] = None, |
153 | 153 | provider: Optional[BaseProvider] = None,
|
154 | 154 | ):
|
155 | 155 | self.__build_config(
|
@@ -223,14 +223,14 @@ def put_metadata(self, key: str, value: Any, namespace: Optional[str] = None):
|
223 | 223 | logger.debug(f"Adding metadata on key '{key}' with '{value}' at namespace '{namespace}'")
|
224 | 224 | self.provider.put_metadata(key=key, value=value, namespace=namespace)
|
225 | 225 |
|
226 |
| - def patch(self, modules: Optional[Tuple[str]] = None): |
| 226 | + def patch(self, modules: Optional[Sequence[str]] = None): |
227 | 227 | """Patch modules for instrumentation.
|
228 | 228 |
|
229 | 229 | Patches all supported modules by default if none are given.
|
230 | 230 |
|
231 | 231 | Parameters
|
232 | 232 | ----------
|
233 |
| - modules : Tuple[str] |
| 233 | + modules : Optional[Sequence[str]] |
234 | 234 | List of modules to be patched, optional by default
|
235 | 235 | """
|
236 | 236 | if self.disabled:
|
@@ -720,7 +720,7 @@ def __build_config(
|
720 | 720 | service: Optional[str] = None,
|
721 | 721 | disabled: Optional[bool] = None,
|
722 | 722 | auto_patch: Optional[bool] = None,
|
723 |
| - patch_modules: Optional[Union[List, Tuple]] = None, |
| 723 | + patch_modules: Optional[Sequence[str]] = None, |
724 | 724 | provider: Optional[BaseProvider] = None,
|
725 | 725 | ):
|
726 | 726 | """Populates Tracer config for new and existing initializations"""
|
|
0 commit comments