def get_uris(self) -> List[str]: # TODO(architkulkarni): this should programmatically be extended with # URIs from all plugins. plugin_uris = [] if "working_dir" in self: plugin_uris.append(self["working_dir"]) if "py_modules" in self: for uri in self["py_modules"]: plugin_uris.append(uri) if "conda" in self: uri = get_conda_uri(self) if uri is not None: plugin_uris.append(uri) if "pip" in self: uri = get_pip_uri(self) if uri is not None: plugin_uris.append(uri)
def _proto_runtime_env(self): if self.__proto_runtime_env: return self.__proto_runtime_env proto_runtime_env = ProtoRuntimeEnv() proto_runtime_env.working_dir = self.get("working_dir", "") if "working_dir" in self: proto_runtime_env.uris.working_dir_uri = self["working_dir"] if "py_modules" in self: proto_runtime_env.python_runtime_env.py_modules.extend( self["py_modules"]) for uri in self["py_modules"]: proto_runtime_env.uris.py_modules_uris.append(uri) if "conda" in self: uri = get_conda_uri(self) if uri is not None: proto_runtime_env.uris.conda_uri = uri if "pip" in self: uri = get_pip_uri(self) if uri is not None: proto_runtime_env.uris.pip_uri = uri env_vars = self.get("env_vars", {}) proto_runtime_env.env_vars.update(env_vars.items()) if "_ray_release" in self: proto_runtime_env.extensions["_ray_release"] = str( self["_ray_release"]) if "_ray_commit" in self: proto_runtime_env.extensions["_ray_commit"] = str( self["_ray_commit"]) if "_inject_current_ray" in self: proto_runtime_env.extensions["_inject_current_ray"] = str( self["_inject_current_ray"]) _build_proto_pip_runtime_env(self, proto_runtime_env) _build_proto_conda_runtime_env(self, proto_runtime_env) _build_proto_container_runtime_env(self, proto_runtime_env) _build_proto_plugin_runtime_env(self, proto_runtime_env) self.__proto_runtime_env = proto_runtime_env return self.__proto_runtime_env
def conda_uri(self) -> Optional[str]: if "conda" in self: return get_conda_uri(self) return None