From a5f538462d0d8faad505a308b060ddfcf5ea2ac7 Mon Sep 17 00:00:00 2001
From: Rafal Blaczkowski <rafal.blaczkowski@intel.com>
Date: Fri, 9 Oct 2020 14:14:10 +0200
Subject: [PATCH] Update OpenVino ONNX CI check (#2599)

* Update OpenVino ONNX CI

* Change parallel execution to single

* Enlarge timeout

* Remove timeout

* Add timeout to test execution
---
 .ci/openvino-onnx/Jenkinsfile | 6 +++++-
 ngraph/python/tox.ini         | 2 +-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/.ci/openvino-onnx/Jenkinsfile b/.ci/openvino-onnx/Jenkinsfile
index 3b6f136fd5f..dd29dc8bc0c 100644
--- a/.ci/openvino-onnx/Jenkinsfile
+++ b/.ci/openvino-onnx/Jenkinsfile
@@ -68,7 +68,7 @@ def buildDockerImage() {
 
 def runTests() {
     sh """
-        docker run --rm --name ${DOCKER_CONTAINER_NAME} \
+        docker run --name ${DOCKER_CONTAINER_NAME} \
         --volume ${HOME}/ONNX_CI/onnx_models/.onnx:/root/.onnx ${DOCKER_IMAGE_TAG}
     """
 }
@@ -101,6 +101,9 @@ pipeline {
             }
         }
         stage("Run tests") {
+            options {
+                  timeout(time: 10, unit: 'MINUTES')
+              }
             steps{
                 runTests()
             }
@@ -118,6 +121,7 @@ pipeline {
                 deleteDir()
                 sh """
                     docker image prune -f
+                    docker rm -f ${DOCKER_CONTAINER_NAME}
                 """
             }
         }
diff --git a/ngraph/python/tox.ini b/ngraph/python/tox.ini
index 8b1de635ab6..de8d2cbaf30 100644
--- a/ngraph/python/tox.ini
+++ b/ngraph/python/tox.ini
@@ -27,7 +27,7 @@ commands=
   flake8 {posargs:src/ setup.py}
   flake8 --ignore=D100,D101,D102,D103,D104,D105,D107,W503 tests/  # ignore lack of docs in tests
   mypy --config-file=tox.ini {posargs:src/}
-  pytest --backend={env:NGRAPH_BACKEND} tests -v -n 20
+  pytest --backend={env:NGRAPH_BACKEND} tests -v -k 'not _cuda' -n 15
 
 [testenv:devenv]
 envdir = devenv