[xray] Fix bug where driver task ID is assumed to be nil (#2725)

## What do these changes do?

#2362 left a bug where it assumed that the driver task ID was nil. This fixes the bug to check the `SchedulingQueue` for any driver task IDs instead.
This commit is contained in:
Stephanie Wang
2018-08-23 14:44:47 -07:00
committed by Robert Nishihara
parent 344a83f327
commit 1b3de31ff1
+2 -4
View File
@@ -1278,10 +1278,8 @@ void NodeManager::HandleObjectMissing(const ObjectID &object_id) {
// Check that remaining tasks that could not be transitioned are running
// workers or drivers, now blocked in a get.
local_queues_.FilterState(waiting_task_id_set, TaskState::RUNNING);
if (!waiting_task_id_set.empty()) {
RAY_CHECK(waiting_task_id_set.size() == 1);
RAY_CHECK(waiting_task_id_set.begin()->is_nil());
}
local_queues_.FilterState(waiting_task_id_set, TaskState::DRIVER);
RAY_CHECK(waiting_task_id_set.empty());
}
}