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())