test fix: None type check (#735)

kubeflow · Jul 18, 2018 · 21f3ea4 · 21f3ea4
1 parent 9e396a4
commit 21f3ea4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py/tf_job_client.py b/py/tf_job_client.py
@@ -97,7 +97,7 @@ def log_status(tf_job):
   """A callback to use with wait_for_job."""
   if tf_job.get("apiVersion", "") == "kubeflow.org/v1alpha2":
     all_conditions = tf_job.get("status", {}).get("conditions", [])
-    conditions = [c.get("type", "") for c in all_conditions]
+    conditions = [] if all_conditions is None else [c.get("type", "") for c in all_conditions]
     logging.info("Job %s in namespace %s; uid=%s; conditions=%s",
                  tf_job.get("metadata", {}).get("name"),
                  tf_job.get("metadata", {}).get("namespace"),