fix ut timeout for tf2.8
@@ -4,7 +4,6 @@ import unittest
import os
import shutil
import yaml
-import tensorflow as tf
import torch
import torchvision
@@ -149,6 +148,7 @@ def build_ox_yaml():
f.close()
def build_fake_model():
+ import tensorflow as tf
try:
graph = tf.Graph()
graph_def = tf.GraphDef()
