Skip to content

Commit 9eb04e2

Browse files
fix: resolve mypy type errors in pe, baseline, and cli
- Add type arguments to bare dict annotations - Explicit str() cast for args.rtmr3 to avoid Any return
1 parent 8e9aed0 commit 9eb04e2

3 files changed

Lines changed: 4 additions & 4 deletions

File tree

src/cvm_measure/cli.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def _resolve_rtmr3(args: argparse.Namespace, parser: argparse.ArgumentParser) ->
118118
rtmr3 = extend_rtmr(bytes(SHA384_SIZE), digest)
119119
return rtmr3.hex()
120120

121-
return args.rtmr3
121+
return str(args.rtmr3) if args.rtmr3 is not None else None
122122

123123

124124
def _require_file(path: Path, flag: str) -> None:
@@ -205,7 +205,7 @@ def _cmd_extract_baseline(args: argparse.Namespace) -> None:
205205
print(f"Baseline written to {args.output}", file=sys.stderr)
206206
else:
207207
from dataclasses import asdict
208-
data: dict = {}
208+
data: dict[str, object] = {}
209209
if baseline.provider:
210210
data["provider"] = baseline.provider
211211
if baseline.platform:

src/cvm_measure/tdx/baseline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ def load(path: Path) -> Baseline:
108108

109109
def save(baseline: Baseline, path: Path) -> None:
110110
"""Save a baseline to a JSON file."""
111-
data: dict = {}
111+
data: dict[str, object] = {}
112112
if baseline.provider:
113113
data["provider"] = baseline.provider
114114
if baseline.platform:

src/cvm_measure/tdx/pe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ def pe_extract_section(
144144
return None
145145

146146

147-
def pe_list_sections(pe_data: bytes) -> list[dict]:
147+
def pe_list_sections(pe_data: bytes) -> list[dict[str, object]]:
148148
"""List all PE sections with metadata."""
149149
if len(pe_data) < 64 or pe_data[:2] != b"MZ":
150150
return []

0 commit comments

Comments
 (0)