Fix null endTime when run status is not finished

Signed-off-by: Timothy Lee <ctiml@infuseai.io>

Author: ctiml

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