Fix timeout in edgeset_test.

PiperOrigin-RevId: 219533560
adology · Oct 31, 2018 · 0f8f062 · 0f8f062
1 parent 26376a4
commit 0f8f062
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorflow/core/graph/edgeset_test.cc b/tensorflow/core/graph/edgeset_test.cc
@@ -15,6 +15,7 @@ limitations under the License.
 
 #include "tensorflow/core/graph/edgeset.h"
 
+#include <set>
 #include <vector>
 #include "tensorflow/core/graph/graph.h"
 #include "tensorflow/core/platform/test.h"
@@ -55,7 +56,7 @@ class EdgeSetTest : public ::testing::Test {
   Edge nonexistent_;
   Edge* edges_;
   EdgeSet eset_;
-  gtl::FlatSet<const Edge*> model_;
+  std::set<const Edge*> model_;
 };
 
 namespace {