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