from logging import getLogger from os import path from typing import Optional from urllib.parse import urlparse from ..utils import ConversionContext from .base import BaseClient logger = getLogger(__name__) class FileClient(BaseClient): protocol = "file://" def __init__(self, _conversion: ConversionContext): """ Nothing to initialize for this client. """ pass def download( self, conversion: ConversionContext, _name: str, uri: str, format: Optional[str] = None, dest: Optional[str] = None, **kwargs, ) -> str: parts = urlparse(uri) logger.info("loading model from: %s", parts.path) return path.join(dest or conversion.model_path, parts.path)