fix: do not mock get (mem0ai#598)

cachho · Sep 12, 2023 · ac08638 · ac08638
1 parent 0314694
commit ac08638
Showing 1 changed file with 0 additions and 2 deletions.
diff --git 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,