Skip to content

Commit d621055

Browse files
committed
added type hinting to builder method returns. removed optional from edge_headers method
1 parent cda21df commit d621055

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

polygon/rest/models/request.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -28,10 +28,10 @@ def __init__(
2828

2929
def edge_headers(
3030
self,
31-
edge_id: Optional[str] = None,
32-
edge_ip_address: Optional[str] = None,
31+
edge_id: str,
32+
edge_ip_address: str,
3333
edge_user: Optional[str] = None,
34-
):
34+
) -> "RequestOptionBuilder":
3535
"""
3636
require_edge_headers adds required headers to the headers' dictionary
3737
:param edge_id: is a required Launchpad header. It identifies the Edge User requesting data
@@ -41,9 +41,6 @@ def edge_headers(
4141
User requesting data
4242
:return ResponseOptionBuilder
4343
"""
44-
if edge_id is None or edge_ip_address is None:
45-
raise RequestOptionError(f"edge_id and edge_ip_address required.")
46-
4744
edge_headers: Dict[str, str] = {
4845
X_POLYGON_EDGE_ID: edge_id,
4946
X_POLYGON_EDGE_IP_ADDRESS: edge_ip_address,
@@ -61,7 +58,7 @@ def update_edge_header(
6158
edge_id: Optional[str] = None,
6259
edge_ip_address: Optional[str] = None,
6360
edge_user: Optional[str] = None,
64-
):
61+
) -> "RequestOptionBuilder":
6562
"""
6663
used to change individual edge elements of underlying headers' dictionary.
6764
:param edge_id: is a required Launchpad header. It identifies the Edge User requesting data

0 commit comments

Comments
 (0)