From 50dc31f362b6d5127b5450994ce25de9704f5d0d Mon Sep 17 00:00:00 2001
From: LiangQuan <lq262469@alibaba-inc.com>
Date: Mon, 9 Sep 2024 12:25:29 +0800
Subject: [PATCH] fix config.toolkit

---
 pai/dsw.py                  | 2 +-
 pai/toolkit/helper/utils.py | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/pai/dsw.py b/pai/dsw.py
index eace7f5..53e797d 100644
--- a/pai/dsw.py
+++ b/pai/dsw.py
@@ -242,7 +242,7 @@ def mount(
 
         resp: GetInstanceResponse = sess._acs_dsw_client.get_instance(self.instance_id)
         datasets = [
-            UpdateInstanceRequestDatasets.from_map(ds.to_map())
+            UpdateInstanceRequestDatasets().from_map(ds.to_map())
             for ds in resp.body.datasets
         ]
         datasets.append(
diff --git a/pai/toolkit/helper/utils.py b/pai/toolkit/helper/utils.py
index 1a93eef..69ef10d 100644
--- a/pai/toolkit/helper/utils.py
+++ b/pai/toolkit/helper/utils.py
@@ -267,12 +267,13 @@ def get_workspace_api(self) -> WorkspaceAPI:
     def get_default_oss_storage_uri(
         self, workspace_id: str
     ) -> Tuple[Optional[str], Optional[str]]:
-        return Session._get_default_oss_storage(
+        bucket_name, endpoint = Session.get_default_oss_storage(
             workspace_id=workspace_id,
             cred=self._get_credential_client(),
             region_id=self.region_id,
             network=self.network,
         )
+        return "oss://{}/".format(bucket_name), endpoint
 
     def set_default_oss_storage(
         self, workspace_id, bucket_name: str, intranet_endpoint: str