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