diff options
Diffstat (limited to 'config-model')
-rw-r--r-- | config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TensorFlowFeatureConverter.java | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TensorFlowFeatureConverter.java b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TensorFlowFeatureConverter.java index 5343d4622c7..01d3449573c 100644 --- a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TensorFlowFeatureConverter.java +++ b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TensorFlowFeatureConverter.java @@ -66,10 +66,10 @@ public class TensorFlowFeatureConverter extends ExpressionTransformer<RankProfil try { ModelStore store = new ModelStore(context.rankProfile().getSearch().sourceApplication(), feature.getArguments()); - if (store.hasTensorFlowModels()) // TODO: Check if we have created a converted model already instead - return transformFromTensorFlowModel(store, context.rankProfile()); - else // is should have previously stored model information instead + if (store.hasStoredModel()) return transformFromStoredModel(store, context.rankProfile()); + else // not converted yet - access TensorFlow model files + return transformFromTensorFlowModel(store, context.rankProfile()); } catch (IllegalArgumentException | UncheckedIOException e) { throw new IllegalArgumentException("Could not use tensorflow model from " + feature, e); @@ -182,12 +182,12 @@ public class TensorFlowFeatureConverter extends ExpressionTransformer<RankProfil public FeatureArguments arguments() { return arguments; } - public boolean hasTensorFlowModels() { + public boolean hasStoredModel() { try { - return application.getFile(ApplicationPackage.MODELS_DIR).exists(); + return application.getFile(arguments.expressionPath()).exists(); } catch (UnsupportedOperationException e) { - return false; // No files -> no TensorFlow models + return false; } } |