From d77b74b41e096af3be970f94672d4e8dcbb0a455 Mon Sep 17 00:00:00 2001 From: KerfuffleV2 Date: Tue, 29 Aug 2023 14:01:05 -0600 Subject: [PATCH] Squish last type warning in gguf.py - yay! --- gguf-py/gguf/gguf.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/gguf-py/gguf/gguf.py b/gguf-py/gguf/gguf.py index c6f4895ac..de3edbc99 100644 --- a/gguf-py/gguf/gguf.py +++ b/gguf-py/gguf/gguf.py @@ -9,6 +9,7 @@ import os from pathlib import Path from enum import IntEnum, auto +from io import BufferedWriter from typing import Any, BinaryIO, Callable, IO, Dict, List, Optional, Sequence, Tuple, Union # @@ -422,7 +423,7 @@ class GGUFValueType(IntEnum): class GGUFWriter: - fout: BinaryIO + fout: BufferedWriter arch: str offset_tensor = 0 data_alignment = GGUF_DEFAULT_ALIGNMENT