From f2b425c59c15ac4621c670aff0578df844d5619a Mon Sep 17 00:00:00 2001 From: brian khuu Date: Thu, 11 Jul 2024 21:52:53 +1000 Subject: [PATCH] convert-*.py: import cast from typing and other refactor --- gguf-py/gguf/metadata.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/gguf-py/gguf/metadata.py b/gguf-py/gguf/metadata.py index e65a72edd..d18ab400e 100644 --- a/gguf-py/gguf/metadata.py +++ b/gguf-py/gguf/metadata.py @@ -4,7 +4,7 @@ import re import json import frontmatter from pathlib import Path -from typing import Any, Optional +from typing import Any, Optional, cast from dataclasses import dataclass from .constants import Keys @@ -69,7 +69,7 @@ class Metadata: metadata.description = metadata_override.get(Keys.General.DESCRIPTION, metadata.description) metadata.quantized_by = metadata_override.get(Keys.General.QUANTIZED_BY, metadata.quantized_by) - metadata.size_label = metadata_override.get(Keys.General.SIZE_LABEL, metadata.size_label) + metadata.size_label = metadata_override.get(Keys.General.SIZE_LABEL, metadata.size_label) metadata.license_name = metadata_override.get(Keys.General.LICENSE_NAME, metadata.license_name) metadata.license_link = metadata_override.get(Keys.General.LICENSE_LINK, metadata.license_link) @@ -105,7 +105,7 @@ class Metadata: return json.load(f) @staticmethod - def load_model_card(model_path: Optional[Path] = None) -> dict[str, object]: + def load_model_card(model_path: Optional[Path] = None) -> dict[str, Any]: if model_path is None or not model_path.exists(): return {} @@ -115,10 +115,10 @@ class Metadata: return {} with open(model_card_path, "r", encoding="utf-8") as f: - return cast("dict[str, object]", frontmatter.load(f)) + return cast("dict[str, Any]", frontmatter.load(f)) @staticmethod - def load_hf_parameters(model_path: Optional[Path] = None) -> dict[str, object]: + def load_hf_parameters(model_path: Optional[Path] = None) -> dict[str, Any]: if model_path is None or not model_path.exists(): return {}