switch to namedtuple, no need to dataclass

This commit is contained in:
Sigbjørn Skjæret 2024-04-21 11:43:43 +02:00
parent bc92f65e8c
commit 87e2d7354f

View file

@ -7,8 +7,7 @@ import json
from pathlib import Path from pathlib import Path
import numpy as np import numpy as np
from typing import Any, Sequence from typing import Any, Sequence, NamedTuple
from dataclasses import dataclass
# Necessary to load the local gguf package # Necessary to load the local gguf package
if "NO_LOCAL_GGUF" not in os.environ and (Path(__file__).parent.parent.parent / 'gguf-py').exists(): if "NO_LOCAL_GGUF" not in os.environ and (Path(__file__).parent.parent.parent / 'gguf-py').exists():
@ -19,8 +18,7 @@ import gguf
logger = logging.getLogger("gguf-new-metadata") logger = logging.getLogger("gguf-new-metadata")
@dataclass class MetadataDetails(NamedTuple):
class MetadataDetails:
type: gguf.GGUFValueType type: gguf.GGUFValueType
value: Any value: Any
description: str = '' description: str = ''