[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_huggingface_embeddings.py
+++ b/tests/embeddings/test_huggingface_embeddings.py
@@ -37,9 +37,7 @@ def test_embed_custom_model(mock_sentence_transformer):


 def test_embed_with_model_kwargs(mock_sentence_transformer):
-    config = BaseEmbedderConfig(
-        model="all-MiniLM-L6-v2", model_kwargs={"device": "cuda"}
-    )
+    config = BaseEmbedderConfig(model="all-MiniLM-L6-v2", model_kwargs={"device": "cuda"})
    embedder = HuggingFaceEmbedding(config)

    mock_sentence_transformer.encode.return_value = [0.7, 0.8, 0.9]