format code

microsoft · Aug 8, 2023 · 9c3147e · 9c3147e
1 parent 4b36cab
commit 9c3147e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cognitive/src/main/python/synapse/ml/cognitive/langchain/LangchainTransform.py b/cognitive/src/main/python/synapse/ml/cognitive/langchain/LangchainTransform.py
@@ -129,7 +129,11 @@ def __init__(
         self.running_on_synapse_internal = running_on_synapse_internal()
         if running_on_synapse_internal():
             from synapse.ml.fabric.service_discovery import get_fabric_env_config
-            self._setDefault(url=get_fabric_env_config().fabric_env_config.ml_workload_endpoint + "cognitive/openai")
+
+            self._setDefault(
+                url=get_fabric_env_config().fabric_env_config.ml_workload_endpoint
+                + "cognitive/openai"
+            )
         kwargs = self._input_kwargs
         if subscriptionKey:
             kwargs["subscriptionKey"] = subscriptionKey
@@ -203,6 +207,7 @@ def udfFunction(x):
 
             if self.running_on_synapse_internal and not self.isSet(self.url):
                 from synapse.ml.fabric.prerun.openai_prerun import OpenAIPrerun
+
                 OpenAIPrerun(api_base=self.getUrl()).init_personalized_session(None)
             else:
                 openai.api_type = "azure"