Fix test 2

h2oai · Mar 29, 2024 · 9c4b3ba · 9c4b3ba
1 parent abced7b
commit 9c4b3ba
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_client_calls.py b/tests/test_client_calls.py
@@ -4932,8 +4932,9 @@ def test_client1_images_qa(langchain_mode, base_model):
                   langchain_mode=langchain_mode,
                   h2ogpt_key=h2ogpt_key)
     res_dict = client.predict(str(dict(kwargs)), api_name='/submit_nochat_api')
+    response = ast.literal_eval(res_dict)['response']
 
-    if base_model in ['liuhaotian/llava-v1.6-vicuna-13b'] and """research paper or academic""" in res:
+    if base_model in ['liuhaotian/llava-v1.6-vicuna-13b'] and """research paper or academic""" in response:
         return
 
     # string of dict for output