rollback tf1 uts

intel-analytics · Sep 1, 2022 · a6d283b · a6d283b
1 parent 7fccc9c
commit a6d283b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 9 deletions.
diff --git a/python/chronos/dev/test/run-pytests.sh b/python/chronos/dev/test/run-pytests.sh
@@ -43,10 +43,10 @@ fi
 
 if [ $RUN_PART1 = 1 ]; then
 echo "Running chronos tests Part 1"
-python -m pytest -v -m "not onnxrt16 and not tf1" test/bigdl/chronos/model \
-                                                  test/bigdl/chronos/forecaster \
-                                                  test/bigdl/chronos/metric \
-                                                  test/bigdl/chronos/pytorch \
+python -m pytest -v -m "not onnxrt16" test/bigdl/chronos/model \
+                                      test/bigdl/chronos/forecaster \
+                                      test/bigdl/chronos/metric \
+                                      test/bigdl/chronos/pytorch \
        -k "not test_forecast_tcmf_distributed"
 exit_status_0=$?
 if [ $exit_status_0 -ne 0 ];
@@ -57,10 +57,10 @@ fi
 
 if [ $RUN_PART2 = 1 ]; then
 echo "Running chronos tests Part 2"
-python -m pytest -v -m "not onnxrt16 and not tf1" test/bigdl/chronos/autots\
-                                                  test/bigdl/chronos/data \
-                                                  test/bigdl/chronos/simulator \
-                                                  test/bigdl/chronos/detector \
+python -m pytest -v -m "not onnxrt16" test/bigdl/chronos/autots\
+                                      test/bigdl/chronos/data \
+                                      test/bigdl/chronos/simulator \
+                                      test/bigdl/chronos/detector \
         -k "not test_ae_fit_score_unrolled"
 exit_status_0=$?
 if [ $exit_status_0 -ne 0 ];

diff --git a/python/chronos/test/bigdl/chronos/__init__.py b/python/chronos/test/bigdl/chronos/__init__.py
@@ -24,6 +24,5 @@
 op_distributed = pytest.mark.distributed
 
 # other mark
-op_tf1 = pytest.mark.tf1 # will delete later, may be in the last pr.
 op_all = pytest.mark.all
 op_onnxrt16 = pytest.mark.onnxrt16