From 9de5466920a54d1c63da793b59e4905c6120a841 Mon Sep 17 00:00:00 2001 From: Francisco Schulz Date: Mon, 13 Feb 2023 16:01:14 +0100 Subject: [PATCH] pipeline caching enabled --- image_prediction/pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/image_prediction/pipeline.py b/image_prediction/pipeline.py index 05984ed..fae919b 100644 --- a/image_prediction/pipeline.py +++ b/image_prediction/pipeline.py @@ -20,7 +20,7 @@ from image_prediction.utils.generic import lift, starlift os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3" -# @lru_cache() +@lru_cache(maxsize=None) def load_pipeline(**kwargs): model_loader = get_mlflow_model_loader(MLRUNS_DIR) model_identifier = CONFIG.service.mlflow_run_id