From d4604a2cb5c688204291b23ea5cc9be3edc4d58a Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Thu, 31 Mar 2022 14:52:37 +0200 Subject: [PATCH] renaming --- image_prediction/default_objects.py | 2 +- scripts/run_pipeline.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/image_prediction/default_objects.py b/image_prediction/default_objects.py index 5580428..d968fef 100644 --- a/image_prediction/default_objects.py +++ b/image_prediction/default_objects.py @@ -11,7 +11,7 @@ def get_mlflow_model_loader(mlruns_dir): return model_loader -def load_up_pipeline(): +def load_pipeline(): model_loader = get_mlflow_model_loader(MLRUNS_DIR) model_identifier = CONFIG.service.run_id diff --git a/scripts/run_pipeline.py b/scripts/run_pipeline.py index 0b22300..90717d9 100644 --- a/scripts/run_pipeline.py +++ b/scripts/run_pipeline.py @@ -1,7 +1,7 @@ import argparse import json -from image_prediction.default_objects import load_up_pipeline +from image_prediction.default_objects import load_pipeline def parse_args(): @@ -14,7 +14,7 @@ def parse_args(): def main(args): - pipeline = load_up_pipeline() + pipeline = load_pipeline() with open(args.pdf, "rb") as f: predictions = pipeline(f.read())