Fix: Verify Batch task has node_info (#681)

Azure · Nov 1, 2018 · 385040d · 385040d
1 parent 6d2e6c5
commit 385040d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/aztk/client/base/helpers/task_table.py b/aztk/client/base/helpers/task_table.py
@@ -40,7 +40,8 @@ def __convert_task_to_entity(partition_key, task):
 def __convert_batch_task_to_aztk_task(batch_task):
     task = Task()
     task.id = batch_task.id
-    task.node_id = batch_task.node_info.node_id
+    if batch_task.node_info:
+        task.node_id = batch_task.node_info.node_id
     task.state = batch_task.state
     task.state_transition_time = batch_task.state_transition_time
     task.command_line = batch_task.command_line