Remove order by clause if it is not needed. Increasing timeout from 10mins to
Reviewed By: asaadaldien
Differential Revision: D6167599
fbshipit-source-id: 3e6bdd55d0aa5b497cc1871f237074b3b9ef6f29
diff --git a/caffe2/python/data_workers.py b/caffe2/python/data_workers.py
index 0e8e870..fdcf882 100644
--- a/caffe2/python/data_workers.py
+++ b/caffe2/python/data_workers.py
@@ -132,7 +132,8 @@
global_coordinator.get_queue(input_source_name, max_buffered_batches),
metrics,
dont_rebatch,
- batch_columns
+ batch_columns,
+ timeout=timeout
)
# Create coordinator object