fix config.toolkit

aliyun · Sep 9, 2024 · 50dc31f · 50dc31f
1 parent b43096d
commit 50dc31f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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