Internal change

PiperOrigin-RevId: 321399730
rally12 · Jul 15, 2020 · b4a9cfe · b4a9cfe
1 parent 615edcf
commit b4a9cfe
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/automl_zero/instruction_test.cc b/automl_zero/instruction_test.cc
@@ -21,12 +21,11 @@
 #include <unordered_set>
 #include <vector>
 
-#include "gtest/gtest.h"
-#include "absl/container/node_hash_set.h"
 #include "definitions.h"
 #include "instruction.pb.h"
 #include "random_generator.h"
 #include "test_util.h"
+#include "gtest/gtest.h"
 
 namespace automl_zero {
 
@@ -89,9 +88,9 @@ Instruction AlterParam(
   return modified_instruction;
 }
 
-absl::node_hash_set<DiffId> Differences(const Instruction& instr1,
-                                        const Instruction& instr2) {
-  absl::node_hash_set<DiffId> differences;
+unordered_set<DiffId> Differences(
+    const Instruction& instr1, const Instruction& instr2) {
+  unordered_set<DiffId> differences;
   if (instr1.op_ != instr2.op_) {
     differences.insert(kDifferentOp);
   }
@@ -126,7 +125,7 @@ absl::node_hash_set<DiffId> Differences(const Instruction& instr1,
 DiffId RandomDifference(
     const Instruction& instr1, const Instruction& instr2) {
   RandomGenerator rand_gen;
-  const absl::node_hash_set<DiffId> differences = Differences(instr1, instr2);
+  const unordered_set<DiffId> differences = Differences(instr1, instr2);
   if (differences.empty()) {
     return kNoDifference;
   }
@@ -138,7 +137,7 @@ DiffId RandomDifference(
 
 IntegerT CountDifferences(
     const Instruction& instr1, const Instruction& instr2) {
-  const absl::node_hash_set<DiffId> differences = Differences(instr1, instr2);
+  const unordered_set<DiffId> differences = Differences(instr1, instr2);
   return differences.size();
 }