From 8a1df76078cb05ded0152ae4a8d8af275a25bc4e Mon Sep 17 00:00:00 2001 From: cdietrich Date: Wed, 2 Mar 2022 10:28:32 +0100 Subject: [PATCH] blacked --- image_prediction/predictor.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/image_prediction/predictor.py b/image_prediction/predictor.py index d994baf..4450e1a 100644 --- a/image_prediction/predictor.py +++ b/image_prediction/predictor.py @@ -27,9 +27,7 @@ class Predictor: """ try: if model_handle is None: - reader = MlflowModelReader( - run_id=CONFIG.service.run_id, mlruns_dir=MLRUNS_DIR - ) + reader = MlflowModelReader(run_id=CONFIG.service.run_id, mlruns_dir=MLRUNS_DIR) self.model_handle = reader.get_model_handle(BASE_WEIGHTS) else: self.model_handle = model_handle