fix linting errors

Author: atulikumwenayo

src/datacustomcode/llm_gateway_config.py

@@ -41,9 +41,7 @@ def to_object(self) -> _E:
 
 
 class LLMGatewayConfig(BaseConfig):
-    llm_gateway_config: Union[
-        LLMGatewayObjectConfig[LLMGateway], None
-    ] = None
+    llm_gateway_config: Union[LLMGatewayObjectConfig[LLMGateway], None] = None
 
     def update(self, other: "LLMGatewayConfig") -> "LLMGatewayConfig":
         def merge(

tests/test_llm_gateway_config_update.py

@@ -19,10 +19,7 @@
 import yaml
 
 from datacustomcode.llm_gateway.default import DefaultLLMGateway
-from datacustomcode.llm_gateway_config import (
-    LLMGatewayConfig,
-    LLMGatewayObjectConfig,
-)
+from datacustomcode.llm_gateway_config import LLMGatewayConfig, LLMGatewayObjectConfig
 
 
 class TestLLMGatewayConfigUpdate:
@@ -61,18 +58,14 @@ def test_update_respects_force_flag(self):
 
         config1.update(config2)
 
-        assert (
-            config1.llm_gateway_config.type_config_name == "ForcedImplementation"
-        )
+        assert config1.llm_gateway_config.type_config_name == "ForcedImplementation"
         assert config1.llm_gateway_config.options == {"forced": True}
         assert config1.llm_gateway_config.force is True
 
 
 class TestLLMGatewayConfigLoad:
     def test_load_from_yaml_file(self):
-        config_data = {
-            "llm_gateway_config": {"type_config_name": "DefaultLLMGateway"}
-        }
+        config_data = {"llm_gateway_config": {"type_config_name": "DefaultLLMGateway"}}
 
         with tempfile.NamedTemporaryFile(mode="w", suffix=".yaml", delete=False) as f:
             yaml.dump(config_data, f)
@@ -83,9 +76,7 @@ def test_load_from_yaml_file(self):
             config.load(temp_file)
 
             assert config.llm_gateway_config is not None
-            assert (
-                config.llm_gateway_config.type_config_name == "DefaultLLMGateway"
-            )
+            assert config.llm_gateway_config.type_config_name == "DefaultLLMGateway"
             llm_gateway = config.llm_gateway_config.to_object()
             assert llm_gateway is not None
             assert isinstance(llm_gateway, DefaultLLMGateway)