summaryrefslogtreecommitdiffstats
path: root/searchlib
diff options
context:
space:
mode:
authorJon Bratseth <bratseth@oath.com>2018-08-10 16:08:54 +0200
committerJon Bratseth <bratseth@oath.com>2018-08-10 16:08:54 +0200
commit66b8b332874136f95fff1290dbd7b7001e4a9398 (patch)
tree51034d282a1ce67e05616a4d545899cff22979cf /searchlib
parentbc41b0e6489e3002d75c400e4dda4f4218306554 (diff)
Refactor and remove duplication
Diffstat (limited to 'searchlib')
-rw-r--r--searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java1
1 files changed, 1 insertions, 0 deletions
diff --git a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java
index ff584559a83..303ba228fa6 100644
--- a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java
+++ b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java
@@ -23,6 +23,7 @@ public class TensorFlowImporter extends ModelImporter {
* @param modelName the name of the model to import, consisting of characters in [A-Za-z0-9_]
* @param modelDir the directory containing the TensorFlow model files to import
*/
+ @Override
public ImportedModel importModel(String modelName, String modelDir) {
try (SavedModelBundle model = SavedModelBundle.load(modelDir, "serve")) {
return importModel(modelName, model);