def deploy(self,
model_version: ModelVersion,
environment_name: str = None,
resource_request: ResourceRequest = None,
env_vars: Dict[str, str] = None) -> VersionEndpoint:
return model_version.deploy(environment_name, resource_request,
env_vars)
def version(project, model, mlflow_url, api_client):
mlflow.set_tracking_uri(mlflow_url)
r = mlflow.start_run()
mlflow.end_run()
v = cl.Version(id=1, model_id=model.id, mlflow_run_id=r.info.run_id, mlflow_url=mlflow_url,
artifact_uri="gs://artifacts",
created_at="2019-08-29T08:13:12.377Z",
updated_at="2019-08-29T08:13:12.377Z")
return ModelVersion(v, model, api_client)
def undeploy(self,
model_version: ModelVersion,
environment_name: str = None):
model_version.undeploy(environment_name)