Patched code sample

import os
from typing import Dict, Optional, TYPE_CHECKING

if TYPE_CHECKING:
    from onnx.onnx_pb import TensorProto  # type: ignore


class ExternalDataInfo:
    """
    A class to hold information about an external data tensor.
    It is a pure data class and does not have any methods.
    It is initialized from a TensorProto.ExternalData protocol buffer message.
    The class has the same fields as the protocol buffer message.

    The main purpose of this class is to make it easier to work with external data
    by providing a more Pythonic interface.
    """

    __slots__ = [
        "location",
        "offset",
        "length",
        "checksum",
        "basepath",
    ]

    def __init__(self, external_data: "TensorProto.ExternalData") -> None:
        self.location: Optional[str] = None
        self.offset: Optional[int] = None
        self.length: Optional[int] = None
        self.checksum: Optional[str] = None
        self.basepath: Optional[str] = None
        for entry in external_data:
            # The use of __slots__ above restricts the attributes that can be set on this class.
            # This prevents a malicious ONNX model from overwriting internal object properties
            # by adding unexpected keys to the external_data field.
            setattr(self, entry.key, entry.value)

    def to_dict(self) -> Dict[str, Optional[str | int]]:
        return {
            "location": self.location,
            "offset": self.offset,
            "length": self.length,
            "checksum": self.checksum,
            "basepath": self.basepath,
        }

    def get_absolute_path(self) -> str:
        if self.basepath and os.path.isabs(self.basepath):
            base_path = self.basepath
        else:
            # If basepath is not absolute, it is relative to the directory of the model.
            # We don't have the model path here, so we assume the current working directory.
            # This is a fallback and might not be correct in all cases.
            base_path = os.getcwd()
            if self.basepath:
                base_path = os.path.join(base_path, self.basepath)

        if not self.location:
            raise ValueError("location is not set")
        return os.path.join(base_path, self.location)