def _get_registry_client(self): """ Attempts to create a py:class:`ModelRegistryClient` if one does not already exist. :raises: py:class:`mlflow.exceptions.MlflowException` if the py:class:`ModelRegistryClient` cannot be created. This may occur, for example, when the registry URI refers to an unsupported store type (e.g., the FileStore). :return: A py:class:`ModelRegistryClient` instance """ # Attempt to fetch a `ModelRegistryClient` that is lazily instantiated and defined as # an instance variable on this `MlflowClient` instance. Because the instance variable # is undefined until the first invocation of _get_registry_client(), the `getattr()` # function is used to safely fetch the variable (if it is defined) or a NoneType # (if it is not defined) registry_client_attr = "_registry_client_lazy" registry_client = getattr(self, registry_client_attr, None) if registry_client is None: try: registry_client = ModelRegistryClient(self._registry_uri) # Define an instance variable on this `MlflowClient` instance to reference the # `ModelRegistryClient` that was just constructed. `setattr()` is used to ensure # that the variable name is consistent with the variable name specified in the # preceding call to `getattr()` setattr(self, registry_client_attr, registry_client) except UnsupportedModelRegistryStoreURIException as exc: raise MlflowException( "Model Registry features are not supported by the store with URI:" " '{uri}'. Stores with the following URI schemes are supported:" " {schemes}.".format(uri=self._registry_uri, schemes=exc.supported_uri_schemes), FEATURE_DISABLED, ) return registry_client
def newModelRegistryClient(): return ModelRegistryClient("uri:/fake")