def __init__(self, project_id=None, region=None, scale_tier=None, job_config=None):
self._project_id = project_id or guess_project_name()
self._region = region or 'us-central1'
self._job_config = job_config or {}
self.scale_tier = scale_tier
self._ml = discovery.build('ml', 'v1')
self._ml._http = http_utils.configure_http_instance(self._ml._http) #pylint:disable=protected-access
def test_configure_http_instance_kwargs():
http = http_utils.configure_http_instance(HTTPMock())
actual = http.request("uri",
"bla",
"blahh",
headers={'user-agent': "uatest"})
assert actual == EXPECTED_UA + " uatest"
def __init__(self, model_dir, model_name, version_name, project_id=None,
**deploy_kwargs):
self._project_id = project_id or guess_project_name()
self._model_dir = model_dir
self._model_name = model_name
self._version_name = version_name
self._deploy_kwargs = deploy_kwargs
self._ml = discovery.build('ml', 'v1')
self._ml._http = http_utils.configure_http_instance(self._ml._http)
# Set default deploy kwargs
if 'runtime_version' not in self._deploy_kwargs:
self._deploy_kwargs['runtime_version'] = '1.13'
if 'python_version' not in self._deploy_kwargs:
self._deploy_kwargs['python_version'] = '3.5'
def test_configure_http_instance_args_no_ua():
http = http_utils.configure_http_instance(HTTPMock())
actual = http.request("uri", "bla", "blahh", {'not_ua': "uatest"})
assert actual == EXPECTED_UA
def test_configure_http_instance_empty_args():
http = http_utils.configure_http_instance(HTTPMock())
actual = http.request()
assert actual == EXPECTED_UA