fix nits

microsoft · Oct 23, 2023 · 6f9c0a1 · 6f9c0a1
1 parent 38a9206
commit 6f9c0a1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/promptflow/promptflow/executor/_tool_resolver.py b/src/promptflow/promptflow/executor/_tool_resolver.py
@@ -249,7 +249,7 @@ def _resolve_llm_node(self, node: Node, convert_input_types=False) -> ResolvedTo
 
     def _resolve_replay_node(self, node: Node, convert_input_types=False) -> ResolvedTool:
         # in replay mode, replace original tool with just_return tool
-        # the tool iteslf just return saved record from storage_record.json
+        # the tool itself just return saved record from storage_record.json
         # processing no logic.
         if (node.api == "completion" or node.api == "chat") and (
             node.connection == "azure_open_ai_connection" or node.provider == "AzureOpenAI"

diff --git a/src/promptflow/tests/sdk_cli_test/unittests/test_pf_client.py b/src/promptflow/tests/sdk_cli_test/unittests/test_pf_client.py
@@ -3,13 +3,15 @@
 # ---------------------------------------------------------
 import mock
 import pytest
-from azure.ai.ml.constants._common import AZUREML_RESOURCE_PROVIDER, RESOURCE_ID_FORMAT
 
 from promptflow import PFClient
 from promptflow._core.operation_context import OperationContext
 from promptflow._sdk.operations._connection_operations import ConnectionOperations
 from promptflow._sdk.operations._local_azure_connection_operations import LocalAzureConnectionOperations
 
+AZUREML_RESOURCE_PROVIDER = "Microsoft.MachineLearningServices"
+RESOURCE_ID_FORMAT = "/subscriptions/{}/resourceGroups/{}/providers/{}/workspaces/{}"
+
 
 @pytest.mark.sdk_test
 @pytest.mark.e2etest