add installation options

intel-analytics · Aug 15, 2022 · f3b355c · f3b355c
1 parent fbff012
commit f3b355c
Showing 1 changed file with 8 additions and 8 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 test/bigdl/chronos/model \
-                    test/bigdl/chronos/forecaster \
-                    test/bigdl/chronos/metric \
-                    test/bigdl/chronos/pytorch \
+python -m pytest -v -m "not tf1 and 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 test/bigdl/chronos/autots\
-                    test/bigdl/chronos/data \
-                    test/bigdl/chronos/simulator \
-                    test/bigdl/chronos/detector \
+python -m pytest -v -m "not tf1 and 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 ];