fix: refactor getOrDefault call for API < 24

spokestack · May 28, 2021 · ddc0ba7 · ddc0ba7
1 parent 62a6ba8
commit ddc0ba7
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/main/java/io/spokestack/spokestack/nlu/tensorflow/WordpieceTextEncoder.java b/src/main/java/io/spokestack/spokestack/nlu/tensorflow/WordpieceTextEncoder.java
@@ -89,8 +89,11 @@ private void loadVocab(String fileName) {
     @Override
     public int encodeSingle(String token) {
         ensureReady();
-        return this.vocabulary.getOrDefault(token,
-              this.vocabulary.get(UNKNOWN));
+        Integer tokenId = this.vocabulary.get(token);
+        if (tokenId == null) {
+            tokenId = this.vocabulary.get(UNKNOWN);
+        }
+        return tokenId;
     }
 
     @Override