diff options
Diffstat (limited to 'src/model_interface_test.cc')
-rw-r--r-- | src/model_interface_test.cc | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/src/model_interface_test.cc b/src/model_interface_test.cc index 52b045d..f5ee492 100644 --- a/src/model_interface_test.cc +++ b/src/model_interface_test.cc @@ -12,11 +12,10 @@ // See the License for the specific language governing permissions and // limitations under the License.! -#include <unordered_map> - #include "model_factory.h" #include "model_interface.h" #include "testharness.h" +#include "third_party/absl/container/flat_hash_map.h" #include "util.h" namespace sentencepiece { @@ -294,8 +293,8 @@ std::string RandomString(int length) { TEST(ModelInterfaceTest, PieceToIdStressTest) { for (const auto type : kModelTypes) { for (int i = 0; i < 100; ++i) { - std::unordered_map<std::string, int> expected_p2i; - std::unordered_map<int, std::string> expected_i2p; + absl::flat_hash_map<std::string, int> expected_p2i; + absl::flat_hash_map<int, std::string> expected_i2p; ModelProto model_proto = MakeBaseModelProto(type); for (int n = 0; n < 1000; ++n) { const std::string piece = RandomString(10); |