From 3599663a22ede158c513e9921ff05a45d49dfcf8 Mon Sep 17 00:00:00 2001 From: Aaron Miller Date: Fri, 30 Jun 2023 16:17:46 -0700 Subject: [PATCH] bindings/python: type assert --- gpt4all-bindings/python/gpt4all/tests/test_gpt4all.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/gpt4all-bindings/python/gpt4all/tests/test_gpt4all.py b/gpt4all-bindings/python/gpt4all/tests/test_gpt4all.py index 983c728e..10ac88fc 100644 --- a/gpt4all-bindings/python/gpt4all/tests/test_gpt4all.py +++ b/gpt4all-bindings/python/gpt4all/tests/test_gpt4all.py @@ -87,7 +87,7 @@ def test_inference_falcon(): model = GPT4All(model_name='ggml-model-gpt4all-falcon-q4_0.bin') prompt = 'hello' output = model.generate(prompt) - + assert isinstance(output, str) assert len(output) > 0 @@ -95,4 +95,5 @@ def test_inference_mpt(): model = GPT4All(model_name='ggml-mpt-7b-chat.bin') prompt = 'hello' output = model.generate(prompt) + assert isinstance(output, str) assert len(output) > 0