From f6c64430007590f5d2b234a7f784e26025d06484 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Mon, 25 Apr 2022 12:18:47 +0200 Subject: [PATCH] renaming --- config.yaml | 2 +- image_prediction/pipeline.py | 2 +- test/fixtures/model_store.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/config.yaml b/config.yaml index 7c88982..6a6111a 100644 --- a/config.yaml +++ b/config.yaml @@ -6,7 +6,7 @@ service: logging_level: $LOGGING_LEVEL_ROOT|INFO # Logging level for service logger verbose: $VERBOSE|True # Service prints document processing progress to stdout batch_size: $BATCH_SIZE|16 # Number of images in memory simultaneously - run_id: $RUN_ID|fabfb1f192c745369b88cab34471aba7 # The ID of the mlflow run to load the service_estimator from + mlflow_run_id: $MLFLOW_RUN_ID|fabfb1f192c745369b88cab34471aba7 # The ID of the mlflow run to load the service_estimator from # These variables control filters that are applied to either images, image metadata or service_estimator predictions. diff --git a/image_prediction/pipeline.py b/image_prediction/pipeline.py index a7accd8..721cbb9 100644 --- a/image_prediction/pipeline.py +++ b/image_prediction/pipeline.py @@ -15,7 +15,7 @@ os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3" def load_pipeline(**kwargs): model_loader = get_mlflow_model_loader(MLRUNS_DIR) - model_identifier = CONFIG.service.run_id + model_identifier = CONFIG.service.mlflow_run_id pipeline = Pipeline(model_loader, model_identifier, **kwargs) diff --git a/test/fixtures/model_store.py b/test/fixtures/model_store.py index 97e1139..49fd639 100644 --- a/test/fixtures/model_store.py +++ b/test/fixtures/model_store.py @@ -46,7 +46,7 @@ def model_loader(database_connector): def mlflow_run_id(): from image_prediction.config import CONFIG - return CONFIG.service.run_id + return CONFIG.service.mlflow_run_id @pytest.fixture