Skip to content
This repository has been archived by the owner on Mar 16, 2024. It is now read-only.

Commit

Permalink
Automatabot run/8 14 2023 (#495)
Browse files Browse the repository at this point in the history
* Diff results test_cli_scripts_run_doc_embedding.py

* Manual fix—missed on instance of variable
  • Loading branch information
NolanTrem committed Aug 23, 2023
1 parent e16023a commit fc86b50
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions automata/tests/unit/cli/test_cli_scripts_run_doc_embedding.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,14 +56,14 @@ def symbol_doc_embedding_handler_mock():
def test_initialize_providers(
get_mock,
set_overrides_mock,
OpenAIEmbeddingProvider_mock,
openai_embedding_provider_mock,
ChromaSymbolEmbeddingVectorDatabase_mock,
SymbolGraph_mock,
symbol_graph_mock,
):
SymbolGraph_mock.return_value = symbol_graph_mock
ChromaSymbolEmbeddingVectorDatabase_mock.return_value = MagicMock()
OpenAIEmbeddingProvider_mock.return_value = MagicMock()
openai_embedding_provider_mock.return_value.return_value = MagicMock()
symbol_code_embedding_handler = MagicMock()
symbol_code_embedding_handler._get_sorted_supported_symbols.return_value = [
FakeSymbol("symbol1"),
Expand Down Expand Up @@ -91,7 +91,7 @@ def test_initialize_providers(
"symbol_graph": symbol_graph_mock,
"code_embedding_db": ChromaSymbolEmbeddingVectorDatabase_mock.return_value,
"doc_embedding_db": ChromaSymbolEmbeddingVectorDatabase_mock.return_value,
"embedding_provider": OpenAIEmbeddingProvider_mock.return_value,
"embedding_provider": openai_embedding_provider_mock.return_value,
"disable_synchronization": True,
}
set_overrides_mock.assert_called_once_with(**overrides)
Expand Down

0 comments on commit fc86b50

Please sign in to comment.