From ac08638a633193677c820edc50c39f06c408112f Mon Sep 17 00:00:00 2001 From: cachho Date: Tue, 12 Sep 2023 19:12:17 +0200 Subject: [PATCH] fix: do not mock get (#598) --- tests/embedchain/test_embedchain.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tests/embedchain/test_embedchain.py b/tests/embedchain/test_embedchain.py index 8122c61a..1bd97e31 100644 --- a/tests/embedchain/test_embedchain.py +++ b/tests/embedchain/test_embedchain.py @@ -10,13 +10,11 @@ class TestChromaDbHostsLoglevel(unittest.TestCase): os.environ["OPENAI_API_KEY"] = "test_key" @patch("chromadb.api.models.Collection.Collection.add") - @patch("chromadb.api.models.Collection.Collection.get") @patch("embedchain.embedchain.EmbedChain.retrieve_from_database") @patch("embedchain.llm.base.BaseLlm.get_answer_from_llm") @patch("embedchain.llm.base.BaseLlm.get_llm_model_answer") def test_whole_app( self, - _mock_get, _mock_add, _mock_ec_retrieve_from_database, _mock_get_answer_from_llm,