From 0d3c3015112d5fee106363fe006aeea1820e8392 Mon Sep 17 00:00:00 2001 From: liangs6212 Date: Wed, 31 Aug 2022 10:07:48 +0800 Subject: [PATCH] rollback tf1 UTs --- python/chronos/test/bigdl/chronos/model/tf2/test_mtnet_keras.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/chronos/test/bigdl/chronos/model/tf2/test_mtnet_keras.py b/python/chronos/test/bigdl/chronos/model/tf2/test_mtnet_keras.py index bfeaf22a2982..4eb043739513 100644 --- a/python/chronos/test/bigdl/chronos/model/tf2/test_mtnet_keras.py +++ b/python/chronos/test/bigdl/chronos/model/tf2/test_mtnet_keras.py @@ -22,6 +22,7 @@ from bigdl.chronos.data import TSDataset import pandas as pd import numpy as np +import tensorflow as tf from numpy.testing import assert_array_almost_equal import tensorflow as tf @@ -56,7 +57,6 @@ def get_data(num_samples): class TestMTNetKeras(TestCase): def setup_method(self, method): - import tensorflow as tf tf.keras.backend.clear_session() train_data, test_data = create_data() self.x_train, y_train = train_data.to_numpy()