rollback tf1 UTs

intel-analytics · Sep 1, 2022 · 0d3c301 · 0d3c301
1 parent a6d283b
commit 0d3c301
Showing 1 changed file with 1 addition and 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
@@ -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()