From 3d1c251e10442dfa7c1c4ddb631af07fd0500507 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Fri, 1 Apr 2022 21:35:10 +0200 Subject: [PATCH] removed redundant TF env var export --- image_prediction/redai_adapter/efficient_net_wrapper.py | 6 +----- image_prediction/redai_adapter/mlflow.py | 1 - image_prediction/redai_adapter/model_wrapper.py | 4 ---- 3 files changed, 1 insertion(+), 10 deletions(-) diff --git a/image_prediction/redai_adapter/efficient_net_wrapper.py b/image_prediction/redai_adapter/efficient_net_wrapper.py index f43a1b7..d35b20c 100644 --- a/image_prediction/redai_adapter/efficient_net_wrapper.py +++ b/image_prediction/redai_adapter/efficient_net_wrapper.py @@ -1,11 +1,7 @@ -import os +import tensorflow as tf from image_prediction.redai_adapter.model_wrapper import ModelWrapper -os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3" - -import tensorflow as tf - class EfficientNetWrapper(ModelWrapper): diff --git a/image_prediction/redai_adapter/mlflow.py b/image_prediction/redai_adapter/mlflow.py index 55ceab3..7930ae3 100644 --- a/image_prediction/redai_adapter/mlflow.py +++ b/image_prediction/redai_adapter/mlflow.py @@ -31,7 +31,6 @@ class MlflowModelReader: @lru_cache(maxsize=None) def __get_run(self, run_id): - return mlflow.get_run(run_id) def __get_classes(self, run_id, prefix="tt"): diff --git a/image_prediction/redai_adapter/model_wrapper.py b/image_prediction/redai_adapter/model_wrapper.py index 7a50f53..c31f4fa 100644 --- a/image_prediction/redai_adapter/model_wrapper.py +++ b/image_prediction/redai_adapter/model_wrapper.py @@ -1,10 +1,6 @@ import abc -import os - -os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3" import numpy as np - import tensorflow as tf