修复由于接口文档与实际返回不对导致的 Bug

baidubce · Nov 14, 2024 · e697c8b · e697c8b
1 parent 79d7fa6
commit e697c8b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/qianfan/model/model.py b/python/qianfan/model/model.py
@@ -210,7 +210,7 @@ def publish(self, name: str = "", **kwargs: Any) -> "Model":
             "taskId": self.task_id,
         }
         if self.step:
-            model_version_meta["step"] = self.step
+            model_version_meta["checkpointStep"] = self.step
         model_publish_resp = ResourceModel.V2.create_custom_model(
             self.set_id,
             console_const.CreateCustomModelSourceType.Train,

diff --git a/python/qianfan/trainer/actions.py b/python/qianfan/trainer/actions.py
@@ -1003,7 +1003,7 @@ def exec(
                 f"ms_{generate_letter_num_random_id(11)}",
                 model_type,
             )
-            existed_model_set_id = model_resp["result"]
+            existed_model_set_id = model_resp["result"]["modelSetId"]
 
         self.model = Model(
             task_id=self.task_id, job_id=self.job_id, set_id=existed_model_set_id