removed redundant TF env var export

This commit is contained in:
Matthias Bisping 2022-04-01 21:35:10 +02:00
parent c80549d5d3
commit 3d1c251e10
3 changed files with 1 additions and 10 deletions

View File

@ -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):

View File

@ -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"):

View File

@ -1,10 +1,6 @@
import abc
import os
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
import numpy as np
import tensorflow as tf