INNER CODE UNIT · Python

_sanitize

onnxsim/onnxsim · onnxsim/rpc/server.py:35

def _sanitize(name: str) -> str:
    base = os.path.basename(name.replace("\\", "/"))
    base = re.sub(r"[^A-Za-z0-9._-]", "_", base).lstrip(".")
    if not base:
        raise proto.RPCError(f"invalid file name {name!r}")
    return base


class _Runner:
    """One loaded model: an onnxruntime session when available, else the reference evaluator."""

    def __init__(
        self,
        model_bytes: bytes,
        providers: Optional[Sequence[str]],
        single_threaded: bool,
    ):
        self.model_bytes = model_bytes

View source record →