`bd7ce7c`

probes/_param_id_mapping: type-ignore on untyped safe_open call

Authored by mfwolffe <wolffemf@dukes.jmu.edu> 2 weeks ago

Status	File	+	-
M	`src/dlm_sway/probes/_param_id_mapping.py`	4	1

src/dlm_sway/probes/_param_id_mapping.pymodified

      # Use ``safe_open`` so we never materialize the tensors — we only
      # need the key list. Saves the 7+ MB read on a typical adapter.
 -    with safe_open(str(safetensors_path), framework="numpy", device="cpu") as fh:
 +    # safetensors ships no py.typed; safe_open is untyped to mypy.
 +    with safe_open(  # type: ignore[no-untyped-call]
 +        str(safetensors_path), framework="numpy", device="cpu"
 +    ) as fh:
          keys = list(fh.keys())
      if not keys:
          raise ParamMappingError(f"{safetensors_path}: no keys found")