diff options
author | Jon Bratseth <bratseth@oath.com> | 2018-08-21 10:35:23 +0200 |
---|---|---|
committer | Jon Bratseth <bratseth@oath.com> | 2018-08-21 10:35:23 +0200 |
commit | 61c3bef3dc00d485ca87cb2e2b145e1b20626bf7 (patch) | |
tree | 52f84d962cbc298535a68b21071e6fced8572e2e | |
parent | b7ede5ee29906e0a1ea61aa42b56913b9d666736 (diff) |
Change to inverted, safer test
2 files changed, 6 insertions, 12 deletions
diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/ConvertedModel.java b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/ConvertedModel.java index 5065c72a368..2bc0ccf6686 100644 --- a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/ConvertedModel.java +++ b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/ConvertedModel.java @@ -5,7 +5,6 @@ import com.yahoo.config.application.api.ApplicationFile; import com.yahoo.config.application.api.ApplicationPackage; import com.yahoo.config.model.application.provider.FilesApplicationPackage; import com.yahoo.io.IOUtils; -import com.yahoo.io.reader.NamedReader; import com.yahoo.path.Path; import com.yahoo.search.query.profile.QueryProfileRegistry; import com.yahoo.searchdefinition.FeatureNames; @@ -82,7 +81,7 @@ public class ConvertedModel { this.modelPath = modelPath; this.modelName = toModelName(modelPath); ModelStore store = new ModelStore(context.rankProfile().getSearch().sourceApplication(), modelPath); - if ( ! store.hasStoredModel()) // not converted yet - access from models/ directory + if ( store.hasSourceModel()) // not converted yet - access from models/ directory expressions = importModel(store, context.rankProfile(), context.queryProfiles(), modelImporter, arguments); else expressions = transformFromStoredModel(store, context.rankProfile()); @@ -93,7 +92,7 @@ public class ConvertedModel { QueryProfileRegistry queryProfiles, ModelImporter modelImporter, FeatureArguments arguments) { - ImportedModel model = modelImporter.importModel(store.modelFiles.modelName(), store.modelDir()); + ImportedModel model = modelImporter.importModel(store.modelFiles.modelName(), store.sourceModelDir()); return transformFromImportedModel(model, store, profile, queryProfiles, arguments); } @@ -501,19 +500,14 @@ public class ConvertedModel { this.modelFiles = new ModelFiles(modelPath); } - public boolean hasStoredModel() { - try { - return application.getFileReference(modelFiles.storedModelPath()).exists(); - } - catch (UnsupportedOperationException e) { - return false; - } + public boolean hasSourceModel() { + return sourceModelDir().exists(); } /** * Returns the directory which contains the source model to use for these arguments */ - public File modelDir() { + public File sourceModelDir() { return application.getFileReference(ApplicationPackage.MODELS_DIR.append(modelFiles.modelPath())); } diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/XgboostFeatureConverter.java b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/XgboostFeatureConverter.java index 62f43e15849..11f64e68519 100644 --- a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/XgboostFeatureConverter.java +++ b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/XgboostFeatureConverter.java @@ -39,7 +39,7 @@ public class XgboostFeatureConverter extends ExpressionTransformer<RankProfileTr ConvertedModel.FeatureArguments arguments = asFeatureArguments(feature.getArguments()); ConvertedModel.ModelStore store = new ConvertedModel.ModelStore(context.rankProfile().getSearch().sourceApplication(), arguments.modelPath()); - RankingExpression expression = xgboostImporter.parseModel(store.modelDir().toString()); + RankingExpression expression = xgboostImporter.parseModel(store.sourceModelDir().toString()); return expression.getRoot(); } catch (IllegalArgumentException | UncheckedIOException e) { throw new IllegalArgumentException("Could not use XGBoost model from " + feature, e); |