modify host_url

Azure · Sep 26, 2024 · 3dbb2d2 · 3dbb2d2
1 parent d688c3c
commit 3dbb2d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_workspace/workspace.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_workspace/workspace.py
@@ -317,7 +317,7 @@ def _load(
         return result
 
     @classmethod
-    def _from_rest_object(cls, rest_obj: RestWorkspace, v2_service_context: Optional[object]) -> Optional["Workspace"]:
+    def _from_rest_object(cls, rest_obj: RestWorkspace, v2_service_context: Optional[object] = None) -> Optional["Workspace"]:
 
         if not rest_obj:
             return None

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_workspace_operations_base.py b/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_workspace_operations_base.py
@@ -95,7 +95,7 @@ def get(self, workspace_name: Optional[str] = None, **kwargs: Any) -> Optional[W
 
         from urllib.parse import urlparse
         parsed_url = urlparse(obj.ml_flow_tracking_uri)
-        host_url = parsed_url.netloc
+        host_url = "https://{}".format(parsed_url.netloc)
         v2_service_context['host_url'] = host_url
 
         # host_url=service_context._get_mlflow_url(),