py : linting with mypy and isort

This commit is contained in:
Jared Van Bortel 2024-01-19 12:38:18 -05:00
parent ffdd051ab5
commit 4a3bc1522e
5 changed files with 15 additions and 12 deletions

View file

@ -5,17 +5,16 @@ import json
import os
import struct
import sys
from pathlib import Path
from typing import Any, BinaryIO, Sequence
import numpy as np
import torch
from pathlib import Path
if 'NO_LOCAL_GGUF' not in os.environ:
sys.path.insert(1, str(Path(__file__).parent / 'gguf-py' / 'gguf'))
import gguf
NUMPY_TYPE_TO_FTYPE: dict[str, int] = {"float32": 0, "float16": 1}