Push Timeout type alias to all relevant locations

stac-utils · Apr 20, 2023 · 3b7c7c7 · 3b7c7c7
1 parent 9ee4166
commit 3b7c7c7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/pystac_client/client.py b/pystac_client/client.py
@@ -7,10 +7,8 @@
     Iterator,
     List,
     Optional,
-    Tuple,
     Union,
 )
-from typing_extensions import TypeAlias
 
 import pystac
 import pystac.utils
@@ -37,15 +35,12 @@
     QueryLike,
     SortbyLike,
 )
-from pystac_client.stac_api_io import StacApiIO
+from pystac_client.stac_api_io import StacApiIO, Timeout
 
 if TYPE_CHECKING:
     from pystac.item import Item as Item_Type
 
 
-Timeout: TypeAlias = Optional[Union[float, Tuple[float, float], Tuple[float, None]]]
-
-
 class Client(pystac.Catalog):
     """A Client for interacting with the root of a STAC Catalog or API
 

diff --git a/pystac_client/stac_api_io.py b/pystac_client/stac_api_io.py
@@ -13,6 +13,7 @@
     Tuple,
     Union,
 )
+from typing_extensions import TypeAlias
 from urllib.parse import urlparse
 
 import pystac
@@ -38,14 +39,17 @@
 logger = logging.getLogger(__name__)
 
 
+Timeout: TypeAlias = Optional[Union[float, Tuple[float, float], Tuple[float, None]]]
+
+
 class StacApiIO(DefaultStacIO):
     def __init__(
         self,
         headers: Optional[Dict[str, str]] = None,
         conformance: Optional[List[str]] = None,
         parameters: Optional[Dict[str, Any]] = None,
         request_modifier: Optional[Callable[[Request], Union[Request, None]]] = None,
-        timeout: Optional[Union[float, Tuple[float, float], Tuple[float, None]]] = None,
+        timeout: Timeout = None,
     ):
         """Initialize class for API IO