removed redundant TF env var export
This commit is contained in:
parent
c80549d5d3
commit
3d1c251e10
@ -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):
|
||||
|
||||
|
||||
@ -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"):
|
||||
|
||||
@ -1,10 +1,6 @@
|
||||
import abc
|
||||
import os
|
||||
|
||||
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
|
||||
|
||||
import numpy as np
|
||||
|
||||
import tensorflow as tf
|
||||
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user