Fix list training_job logs.

Signed-off-by: pitt-liang <[email protected]>
aliyun · Jan 4, 2024 · 06d5bac · 06d5bac
1 parent 1a7f0c2
commit 06d5bac
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/pai/api/training_job.py b/pai/api/training_job.py
@@ -186,4 +186,10 @@ def list_logs(
             training_job_id=training_job_id,
             request=request,
         )
-        return self.make_paginated_result(resp)
+        # resp.logs may be None
+        logs = resp.logs or []
+        total_count = resp.total_count or 0
+        return PaginatedResult(
+            items=logs,
+            total_count=total_count,
+        )
diff --git a/pai/version.py b/pai/version.py
@@ -12,4 +12,4 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
-VERSION = "0.4.5"
+VERSION = "0.4.5.post0"