From c144c9689b4addbd7b5f9af1e75ca3f6ba179aed Mon Sep 17 00:00:00 2001 From: Li Wan Date: Tue, 26 Mar 2024 16:21:37 +1100 Subject: [PATCH] Fix unit tests errors --- tests/tensor_search/integ_tests/test_search_unstructured.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/tensor_search/integ_tests/test_search_unstructured.py b/tests/tensor_search/integ_tests/test_search_unstructured.py index 8b743d242..518035ffb 100644 --- a/tests/tensor_search/integ_tests/test_search_unstructured.py +++ b/tests/tensor_search/integ_tests/test_search_unstructured.py @@ -6,6 +6,7 @@ from unittest import mock import requests +from pydantic import ValidationError import marqo.core.exceptions as core_exceptions from marqo.api import exceptions as errors @@ -1080,7 +1081,7 @@ def test_multi_search_images_invalid_queries(self): {"https://marqo_not_real.com/image_1.png": 3}, set()] for q in invalid_queries: with self.subTest(f"query={q}"): - with self.assertRaises(errors.InvalidArgError): + with self.assertRaises((ValidationError, errors.InvalidArgError)) as e: tensor_search.search( text=q, index_name=self.default_image_index,