@@ -132,18 +132,15 @@ def get_oss_model_url(model_full_path):
132132 return "oss://%s/%s" % (oss .SQLFLOW_MODELS_BUCKET , model_full_path )
133133
134134
135- def parse_maxcompute_dsn (datasource ):
136- return MaxComputeConnection .get_uri_parts (datasource )
137-
138-
139135def drop_pai_model (datasource , model_name ):
140136 """Drop PAI model
141137
142138 Args:
143139 datasource: current datasource
144140 model_name: name of the model to drop
145141 """
146- user , passwd , address , database = parse_maxcompute_dsn (datasource )
142+ user , passwd , address , database = MaxComputeConnection .get_uri_parts (
143+ datasource )
147144 cmd = "drop offlinemodel if exists %s" % model_name
148145 subprocess .run ([
149146 "odpscmd" , "-u" , user , "-p" , passwd , "--project" , database ,
@@ -220,7 +217,8 @@ def submit_pai_task(pai_cmd, datasource):
220217 pai_cmd: The command to submit
221218 datasource: The datasource this cmd will manipulate
222219 """
223- user , passwd , address , project = parse_maxcompute_dsn (datasource )
220+ user , passwd , address , project = MaxComputeConnection .get_uri_parts (
221+ datasource )
224222 cmd = [
225223 "odpscmd" , "--instance-priority" , "9" , "-u" , user , "-p" , passwd ,
226224 "--project" , project , "--endpoint" , address , "-e" , pai_cmd
@@ -234,22 +232,18 @@ def submit_pai_task(pai_cmd, datasource):
234232def get_oss_model_save_path (datasource , model_name ):
235233 if not model_name :
236234 return None
237- user , _ , _ , project = parse_maxcompute_dsn (datasource )
235+ user , _ , _ , project = MaxComputeConnection . get_uri_parts (datasource )
238236 user = user or "unknown"
239237 return "/" .join ([project , user , model_name ])
240238
241239
242- def get_datasource_dsn (datasource ):
243- return datasource .split ("://" )[1 ]
244-
245-
246240def get_project (datasource ):
247241 """Get the project info from given datasource
248242
249243 Args:
250244 datasource: The odps url to extract project
251245 """
252- _ , _ , _ , project = parse_maxcompute_dsn (datasource )
246+ _ , _ , _ , project = MaxComputeConnection . get_uri_parts (datasource )
253247 return project
254248
255249
0 commit comments