[Misc] Lint code and fix code smells (#1871)

2024-09-16 17:39:54 -07:00
parent 0a78cb9f7a
commit 55c54beeab
57 changed files with 1178 additions and 1357 deletions
--- a/tests/embeddings/test_ollama_embeddings.py
+++ b/tests/embeddings/test_ollama_embeddings.py
@@ -23,9 +23,7 @@ def test_embed_text(mock_ollama_client):
    text = "Sample text to embed."
    embedding = embedder.embed(text)

-    mock_ollama_client.embeddings.assert_called_once_with(
-        model="nomic-embed-text", prompt=text
-    )
+    mock_ollama_client.embeddings.assert_called_once_with(model="nomic-embed-text", prompt=text)

    assert embedding == [0.1, 0.2, 0.3, 0.4, 0.5]