[openvino_dev] Added clean-up files in "<build dir>/tools/openvino_dev"

RaguzovaTatyana · Aug 10, 2023 · 6a977f6 · 6a977f6
1 parent dab01d2
commit 6a977f6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tools/openvino_dev/setup.py b/tools/openvino_dev/setup.py
@@ -27,6 +27,7 @@
 SCRIPT_DIR = Path(__file__).resolve().parents[0]
 OPENVINO_DIR = Path(__file__).resolve().parents[2]
 SRC_DIR = SCRIPT_DIR / 'src'
+WORKING_DIR = Path.cwd()
 
 PKG_INSTALL_CFG = {
     'openvino-mo': {
@@ -178,6 +179,7 @@ def clean_temp_files(self):
                 src_dir = Path(comp_data.get('src_dir'))
                 paths += src_dir.glob(pattern)
             paths += SCRIPT_DIR.glob(pattern)
+            paths += os.path.join(WORKING_DIR, "build", "tools", "openvino_dev").glob(pattern)
             for path in paths:
                 if path.is_file() and path.exists():
                     path = path.parent
@@ -308,7 +310,7 @@ def concat_files(output_file, input_files):
         'install': CustomInstall,
         'clean': CustomClean,
     },
-    entry_points = {
+    entry_points={
         'console_scripts': [],
     },
     install_requires=read_requirements(SCRIPT_DIR / 'requirements.txt'),