[openvino_dev] Added new extra requirements for diff tasks

RaguzovaTatyana · Sep 1, 2023 · c04a21b · c04a21b
1 parent 20bf7ae
commit c04a21b
Show file tree

Hide file tree

Showing 6 changed files with 24 additions and 1 deletion.
diff --git a/tools/openvino_dev/CMakeLists.txt b/tools/openvino_dev/CMakeLists.txt
@@ -36,7 +36,7 @@ set(WHEEL_BUILD "${OpenVINO_VERSION_BUILD}" CACHE STRING "Build number of this r
 ov_cpack_add_component(${OV_CPACK_COMP_OPENVINO_DEV_REQ_FILES} HIDDEN)
 
 set(REQUIREMENTS_IN "${CMAKE_CURRENT_SOURCE_DIR}/requirements_dev.txt.in")
-set(EXTRAS_LIST _ caffe kaldi mxnet onnx pytorch tensorflow tensorflow2)
+set(EXTRAS_LIST _ caffe kaldi mxnet onnx pytorch tensorflow tensorflow2 nlp vision)
 
 foreach(EXTRAS IN LISTS EXTRAS_LIST)
     if(EXTRAS STREQUAL "_")

diff --git a/tools/openvino_dev/requirements_.txt b/tools/openvino_dev/requirements_.txt
@@ -0,0 +1 @@
+-c ../constraints.txt
diff --git a/tools/openvino_dev/requirements_nlp.txt b/tools/openvino_dev/requirements_nlp.txt
@@ -0,0 +1,8 @@
+-c ../constraints.txt
+tensorflow<2.13.0
+defusedxml
+protobuf
+PyYAML
+requests
+tqdm
+urllib3
diff --git a/tools/openvino_dev/requirements_sr.txt b/tools/openvino_dev/requirements_sr.txt
@@ -0,0 +1 @@
+-c ../constraints.txt
diff --git a/tools/openvino_dev/requirements_vision.txt b/tools/openvino_dev/requirements_vision.txt
@@ -0,0 +1,9 @@
+-c ../constraints.txt
+tensorflow<2.13.0
+torchvision
+opencv-python
+protobuf
+PyYAML
+requests
+tqdm
+urllib3
diff --git a/tools/openvino_dev/setup.py b/tools/openvino_dev/setup.py
@@ -312,6 +312,10 @@ def concat_files(output_file, input_files):
         'console_scripts': [],
     },
     install_requires=read_requirements(SCRIPT_DIR / 'requirements.txt'),
+    extras_require={
+        'nlp': read_requirements(SCRIPT_DIR / 'requirements_nlp.txt'),
+        'vision': read_requirements(SCRIPT_DIR / 'requirements_vision.txt'),
+    },
     packages=find_namespace_packages(where=str(SRC_DIR)),
     package_dir={'': str(SRC_DIR)},
 )