Code Formatting (#1828)

embedchain/tests/embedder/test_azure_openai_embedder.py

@@ -1,4 +1,4 @@
-from unittest.mock import patch, Mock
+from unittest.mock import Mock, patch
 
 import httpx
 

embedchain/tests/embedder/test_huggingface_embedder.py

@@ -1,5 +1,6 @@
 
 from unittest.mock import patch
+
 from embedchain.config import BaseEmbedderConfig
 from embedchain.embedder.huggingface import HuggingFaceEmbedder
 

embedchain/tests/helper_classes/test_json_serializable.py

@@ -4,8 +4,10 @@ from string import Template
 
 from embedchain import App
 from embedchain.config import AppConfig, BaseLlmConfig
-from embedchain.helpers.json_serializable import (JSONSerializable,
-                                                  register_deserializable)
+from embedchain.helpers.json_serializable import (
+    JSONSerializable,
+    register_deserializable,
+)
 
 
 class TestJsonSerializable(unittest.TestCase):

embedchain/tests/llm/conftest.py

@@ -1,7 +1,9 @@
 
 from unittest import mock
+
 import pytest
 
+
 @pytest.fixture(autouse=True)
 def mock_alembic_command_upgrade():
     with mock.patch("alembic.command.upgrade"):

embedchain/tests/llm/test_azure_openai.py

@@ -1,4 +1,4 @@
-from unittest.mock import Mock, MagicMock, patch
+from unittest.mock import MagicMock, Mock, patch
 
 import httpx
 import pytest

embedchain/tests/llm/test_ollama.py

@@ -1,8 +1,8 @@
 import pytest
+from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 
 from embedchain.config import BaseLlmConfig
 from embedchain.llm.ollama import OllamaLlm
-from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 
 
 @pytest.fixture

embedchain/tests/models/test_data_type.py

@@ -1,5 +1,9 @@
-from embedchain.models.data_type import (DataType, DirectDataType,
-                                         IndirectDataType, SpecialDataType)
+from embedchain.models.data_type import (
+    DataType,
+    DirectDataType,
+    IndirectDataType,
+    SpecialDataType,
+)
 
 
 def test_subclass_types_in_data_type():