misc: pylint fixes

Author: muhammadyaseen

graphgen/models/llm/api/openai_client.py

@@ -68,9 +68,9 @@ def __init__(
 
     def __post_init__(self):
 
-        api_name = self.backend.replace("_", " ") 
+        api_name = self.backend.replace("_", " ")
         assert self.api_key is not None, f"Please provide api key to access {api_name}."
-        if self.backend == "openai_api":    
+        if self.backend == "openai_api":
             self.client = AsyncOpenAI(
                 api_key=self.api_key or "dummy", base_url=self.base_url
             )

graphgen/operators/init/init_llm.py

@@ -29,7 +29,7 @@ def create_llm_wrapper(backend: str, config: Dict[str, Any]) -> BaseLLMWrapper:
             return HTTPClient(**config)
         if backend in ("openai_api", "azure_openai_api"):
             from graphgen.models.llm.api.openai_client import OpenAIClient
-            # pass in concrete backend to the OpenAIClient so that internally we can distinguish 
+            # pass in concrete backend to the OpenAIClient so that internally we can distinguish
             # between OpenAI and Azure OpenAI
             return OpenAIClient(**config, backend=backend)
         if backend == "ollama_api":