feat: support model compression spec (#42)

* update aiworkspace lib * update model recipe default output channels format with oss endpoint * add test case for model compression Co-authored-by: yutou.zy <[email protected]>
aliyun · Oct 9, 2024 · 4d9d3ef · 4d9d3ef
1 parent 9649fc4
commit 4d9d3ef
Show file tree

Hide file tree

Showing 7 changed files with 8,900 additions and 3,233 deletions.
diff --git a/pai/api/model.py b/pai/api/model.py
@@ -77,6 +77,7 @@ def create(
 
     def list(
         self,
+        collections: str = None,
         domain: str = None,
         label: str = None,
         label_string: str = None,
@@ -93,6 +94,7 @@ def list(
         workspace_id: str = None,
     ) -> PaginatedResult:
         request = ListModelsRequest(
+            collections=collections,
             domain=domain,
             label=label,
             label_string=label_string,

diff --git a/pai/job/_training_job.py b/pai/job/_training_job.py
@@ -708,7 +708,10 @@ def _training_job_base_output(self, job_name):
             StoragePathCategory.TrainingJob,
             f"{to_plain_text(job_name)}_{random_str(6)}",
         )
-        base_output_path = f"oss://{bucket_name}/{storage_path}"
+        base_output_path = (
+            f"oss://{bucket_name}.{session.oss_endpoint}/"
+            f"{storage_path}"
+        )
         return base_output_path
 
     def build_outputs(

diff --git a/pai/libs/alibabacloud_aiworkspace20210204/__init__.py b/pai/libs/alibabacloud_aiworkspace20210204/__init__.py
@@ -1 +1 @@
-__version__ = '3.0.2'
+__version__ = '3.0.6'