From 4ae03a0d5dda6eab2cc9e68c806c77c4f3cb9c0e Mon Sep 17 00:00:00 2001
From: Artyom Anokhov <artyom.anokhov@intel.com>
Date: Mon, 27 Jul 2020 19:45:26 +0300
Subject: [PATCH] setupvars: Added post_training_optimization_toolkit to
 PYTHONPATH (#1492)

---
 scripts/setupvars/setupvars.bat | 4 ++++
 scripts/setupvars/setupvars.sh  | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/scripts/setupvars/setupvars.bat b/scripts/setupvars/setupvars.bat
index 1f08c1480cc..88251e850c9 100644
--- a/scripts/setupvars/setupvars.bat
+++ b/scripts/setupvars/setupvars.bat
@@ -101,6 +101,10 @@ if exist %INTEL_OPENVINO_DIR%\deployment_tools\open_model_zoo\tools\accuracy_che
     set PYTHONPATH=%INTEL_OPENVINO_DIR%\deployment_tools\open_model_zoo\tools\accuracy_checker;%PYTHONPATH%
 )
 
+if exist %INTEL_OPENVINO_DIR%\deployment_tools\tools\post_training_optimization_toolkit (
+    set PYTHONPATH=%INTEL_OPENVINO_DIR%\deployment_tools\tools\post_training_optimization_toolkit;%PYTHONPATH%
+)
+
 echo [setupvars.bat] OpenVINO environment initialized
 
 exit /B 0
diff --git a/scripts/setupvars/setupvars.sh b/scripts/setupvars/setupvars.sh
index 4682914e5c1..1739be95dd0 100755
--- a/scripts/setupvars/setupvars.sh
+++ b/scripts/setupvars/setupvars.sh
@@ -94,6 +94,10 @@ if [ -e $INTEL_OPENVINO_DIR/deployment_tools/open_model_zoo/tools/accuracy_check
     export PYTHONPATH="$INTEL_OPENVINO_DIR/deployment_tools/open_model_zoo/tools/accuracy_checker:$PYTHONPATH"
 fi
 
+if [ -e $INTEL_OPENVINO_DIR/deployment_tools/tools/post_training_optimization_toolkit ]; then
+    export PYTHONPATH="$INTEL_OPENVINO_DIR/deployment_tools/tools/post_training_optimization_toolkit:$PYTHONPATH"
+fi
+
 if [ -z "$python_version" ]; then
     if command -v python3.7 >/dev/null 2>&1; then
         python_version=3.7