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 __init__(self,
registry=None,
image_name=None,
base_image=constants.DEFAULT_BASE_IMAGE,
push=True,
preprocessor=None,
dockerfile_path=None):
self.registry = registry
self.image_name = image_name
self.push = push
if self.registry is None:
# TODO(r2d4): Add more heuristics here...
# If no push and no registry provided, use any registry name
if not self.push:
self.registry = 'local/fairing-image'
else:
self.registry = 'gcr.io/{}'.format(gcp.guess_project_name())
self.base_image = base_image
self.dockerfile_path = dockerfile_path
self.preprocessor = preprocessor
self.image_tag = None
self.docker_client = None
def test_guess_project_name_application_default_file(tmp_path):
creds_file = tmp_path / 'credentials'
project_id = 'test_project'
with creds_file.open('w') as f:
json.dump({'project_id': project_id}, f)
assert guess_project_name(str(creds_file)) == project_id
def __init__(self, project_id=None):
self._project_id = project_id or guess_project_name()
self._ml = discovery.build('ml', 'v1')
# Set default deploy kwargs
self._deploy_kwargs = {
'runtime_version': '1.12',
'python_version': '3.5'
}
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')
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 __init__(self,
registry=None,
base_image=constants.DEFAULT_BASE_IMAGE,
preprocessor=None,
dockerfile_path=None):
self.registry = registry
if self.registry is None:
# TODO(r2d4): Add more heuristics here...
self.registry = 'gcr.io/{}'.format(gcp.guess_project_name())
self.base_image = base_image
self.dockerfile_path = dockerfile_path
self.preprocessor = preprocessor
self.image_name = None
self.image_tag = None
self.docker_client = None
if self.registry.count("/") == 0:
self.registry = "{DEFAULT_REGISTRY}/{USER_REPOSITORY}".format(
DEFAULT_REGISTRY=constants.DEFAULT_REGISTRY,
USER_REPOSITORY=self.registry)
def prepare(self, context_filename):
if self.gcp_project is None:
self.gcp_project = gcp.guess_project_name()
self.uploaded_context_url = self.upload_context(context_filename)
self.created_secret = self.create_secret()
def test_guess_project_name_google_auth(tmp_path):
project_id = 'test_project'
with patch('google.auth.default', return_value=(None, project_id)):
assert guess_project_name() == project_id
def __init__(self, project_id=None, region=None, scale_tier="BASIC"):
self._project_id = project_id or guess_project_name()
self._region = region or 'us-central1'
self.scale_tier = scale_tier
self._ml = discovery.build('ml', 'v1')
def __init__(self, project_id=None, region=None, training_scale_tier=None):
super(GCPManagedBackend, self).__init__()
self._project_id = project_id or gcp.guess_project_name()
self._region = region or 'us-central1'
self._training_scale_tier = training_scale_tier or 'BASIC'
def test_guess_project_name_google_auth(tmp_path): #pylint:disable=unused-argument
project_id = 'test_project'
with patch('google.auth.default', return_value=(None, project_id)):
assert guess_project_name() == project_id
def prepare(self, context_filename): # pylint:disable=arguments-differ
if self.gcp_project is None:
self.gcp_project = gcp.guess_project_name()
self.uploaded_context_url = self.upload_context(context_filename)
