gguf-py : fix flake8 lint

This commit is contained in:
Francis Couture-Harpin 2024-07-17 23:26:45 -04:00
parent 2c18a9a4d4
commit 4c9932c1e1

View file

@ -11,6 +11,7 @@ if "NO_LOCAL_GGUF" not in os.environ and (Path(__file__).parent.parent.parent /
import gguf import gguf
class TestMetadataMethod(unittest.TestCase): class TestMetadataMethod(unittest.TestCase):
def test_id_to_title(self): def test_id_to_title(self):
@ -116,5 +117,6 @@ class TestMetadataMethod(unittest.TestCase):
expect = gguf.Metadata(name='Hermes 2 Pro Llama 3 8b DPO', finetune='DPO', basename='hermes-2-pro-llama-3', size_label='8B') expect = gguf.Metadata(name='Hermes 2 Pro Llama 3 8b DPO', finetune='DPO', basename='hermes-2-pro-llama-3', size_label='8B')
self.assertEqual(got, expect) self.assertEqual(got, expect)
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main()