diff --git a/gguf-py/tests/test_codegen_bias.py b/gguf-py/tests/test_codegen_bias.py index 9bb0507429403..3468712e44b9c 100644 --- a/gguf-py/tests/test_codegen_bias.py +++ b/gguf-py/tests/test_codegen_bias.py @@ -1,17 +1,21 @@ -import os + import json +import os import tempfile import unittest +from pathlib import Path import pytest + pytest.importorskip("torch") -import torch -import numpy as np -import gguf -from pathlib import Path +import torch # noqa: E402 +import gguf # noqa: E402 + +from convert_hf_to_gguf import CodeGenModel # noqa: E402 + + -from convert_hf_to_gguf import CodeGenModel class TestCodeGenBias(unittest.TestCase): def _create_model(self, dir_model: str, with_bias: bool):