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)