Merge pull request #107 from InfuseAI/bug/fix-list-runs-without-endtime

Fix null endTime when run status is not finished
InfuseAI · Dec 29, 2022 · 62ad7a5 · 62ad7a5
2 parents cc51a72 + d6c6803
commit 62ad7a5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/primehub/models.py b/primehub/models.py
@@ -226,8 +226,9 @@ def list_runs(self, experiment_name: str):
                     'experimentId': r['info']['experimentId'],
                     'status': r['info']['status'],
                     'startTime': timestamp_to_isoformat(r['info']['startTime']),
-                    'endTime': timestamp_to_isoformat(r['info']['endTime']),
                 }
+                if r['info'].get('endTime'):
+                    m['endTime'] = timestamp_to_isoformat(r['info']['endTime'])
                 yield m
         return results