From 06d5bac94190d4ca5e33f189620c6b9f32fac23e Mon Sep 17 00:00:00 2001
From: pitt-liang <lq262469@alibaba-inc.com>
Date: Thu, 4 Jan 2024 13:52:28 +0800
Subject: [PATCH] Fix list training_job logs.

Signed-off-by: pitt-liang <lq262469@alibaba-inc.com>
---
 pai/api/training_job.py | 8 +++++++-
 pai/version.py          | 2 +-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/pai/api/training_job.py b/pai/api/training_job.py
index 64dcc91..4f56b02 100644
--- 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
index 16a1245..7d4c2a5 100644
--- 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"