support label in model deploy

aliyun · Apr 25, 2024 · d77859e · d77859e
1 parent d8e2333
commit d77859e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/pai/model.py b/pai/model.py
@@ -832,6 +832,7 @@ def deploy(
                 options=options,
                 wait=wait,
                 serializer=serializer,
+                **kwargs,
             )
 
     def _generate_service_name(self):
@@ -849,6 +850,7 @@ def _deploy(
         options: Dict[str, Any] = None,
         wait: bool = True,
         serializer: "SerializerBase" = None,
+        labels: Optional[Dict[str, str]] = None,
     ):
         """Create a prediction service."""
         if not service_name:
@@ -867,7 +869,7 @@ def _deploy(
             resource_id=resource_id,
             options=options,
         )
-        service_name = self.session.service_api.create(config=config)
+        service_name = self.session.service_api.create(config=config, labels=labels)
         self._wait_service_visible(service_name)
         if service_type == ServiceType.Async:
             predictor = AsyncPredictor(
@@ -1738,6 +1740,7 @@ def deploy(
                 options=options,
                 wait=wait,
                 serializer=serializer,
+                **kwargs,
             )
 
     def _build_service_config(

diff --git a/pai/modelscope/model.py b/pai/modelscope/model.py
@@ -350,4 +350,5 @@ def deploy(
             options=options,
             wait=wait,
             serializer=serializer,
+            **kwargs,
         )