gguf-py : fix flake8 lint
This commit is contained in:
parent
2c18a9a4d4
commit
4c9932c1e1
1 changed files with 4 additions and 2 deletions
|
@ -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()
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue