This commit is contained in:
Matheus Garvão 2023-07-17 21:37:59 -05:00 committed by GitHub
commit 9fb9eefe1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -422,8 +422,8 @@ public class EarlybirdWireModule {
public ScoringModelsManager provideScoringModelsManager( public ScoringModelsManager provideScoringModelsManager(
SearchStatsReceiver serverStats, SearchStatsReceiver serverStats,
EarlybirdIndexConfig earlybirdIndexConfig) { EarlybirdIndexConfig earlybirdIndexConfig) {
boolean modelsEnabled = EarlybirdConfig.getBool("scoring_models_enabled", false); boolean modelsDisabled = !EarlybirdConfig.getBool("scoring_models_enabled", false);
if (!modelsEnabled) { if (modelsDisabled) {
LOG.info("Scoring Models - Disabled in the config. Not loading any models."); LOG.info("Scoring Models - Disabled in the config. Not loading any models.");
serverStats.getCounter("scoring_models_disabled_in_config").increment(); serverStats.getCounter("scoring_models_disabled_in_config").increment();
return ScoringModelsManager.NO_OP_MANAGER; return ScoringModelsManager.NO_OP_MANAGER;
@ -450,9 +450,9 @@ public class EarlybirdWireModule {
Decider decider, Decider decider,
EarlybirdIndexConfig earlybirdIndexConfig) { EarlybirdIndexConfig earlybirdIndexConfig) {
boolean modelsEnabled = EarlybirdProperty.TF_MODELS_ENABLED.get(false); boolean modelsDisabled = !EarlybirdProperty.TF_MODELS_ENABLED.get(false);
if (!modelsEnabled) { if (modelsDisabled) {
LOG.info("Tensorflow Models - Disabled in the config. Not loading any models."); LOG.info("Tensorflow Models - Disabled in the config. Not loading any models.");
serverStats.getCounter("tf_models_disabled_in_config").increment(); serverStats.getCounter("tf_models_disabled_in_config").increment();
return TensorflowModelsManager.createNoOp(statsPrefix); return TensorflowModelsManager.createNoOp(statsPrefix);