diff --git a/yarn_api_client/base.py b/yarn_api_client/base.py index 999f78f..d61fd42 100644 --- a/yarn_api_client/base.py +++ b/yarn_api_client/base.py @@ -58,7 +58,7 @@ def to_url(self, api_path=None): class BaseYarnAPI(object): response_class = Response - def __init__(self, service_endpoint=None, timeout=None, auth=None, verify=True): + def __init__(self, service_endpoint=None, timeout=None, auth=None, verify=True, proxies=None): self.timeout = timeout if service_endpoint: @@ -69,6 +69,7 @@ def __init__(self, service_endpoint=None, timeout=None, auth=None, verify=True): self.session = requests.Session() self.session.auth = auth self.session.verify = verify + self.session.proxies = proxies def _validate_configuration(self): if not self.service_uri: