diff options
author | Jon Bratseth <bratseth@oath.com> | 2018-08-10 16:08:54 +0200 |
---|---|---|
committer | Jon Bratseth <bratseth@oath.com> | 2018-08-10 16:08:54 +0200 |
commit | 66b8b332874136f95fff1290dbd7b7001e4a9398 (patch) | |
tree | 51034d282a1ce67e05616a4d545899cff22979cf /searchlib | |
parent | bc41b0e6489e3002d75c400e4dda4f4218306554 (diff) |
Refactor and remove duplication
Diffstat (limited to 'searchlib')
-rw-r--r-- | searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java | 1 |
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); |