diff --git a/python/setup_onnxruntime.py b/python/setup_onnxruntime.py
index 944235f..b103b55 100644
--- a/python/setup_onnxruntime.py
+++ b/python/setup_onnxruntime.py
@@ -54,7 +54,7 @@ setuptools.setup(
     include_package_data=True,
     install_requires=["pyclipper>=1.2.1", "onnxruntime>=1.7.0",
                       "opencv_python>=4.5.1.48", "numpy>=1.19.3",
-                      "six>=1.15.0", "Shapely>=1.7.1"],
+                      "six>=1.15.0", "Shapely>=1.7.1", 'PyYAML'],
     package_dir={'': MODULE_NAME},
     packages=setuptools.find_namespace_packages(where=MODULE_NAME),
     package_data={'': ['*.onnx', '*.yaml']},
@@ -66,5 +66,6 @@ setuptools.setup(
         'Programming Language :: Python :: 3.7',
         'Programming Language :: Python :: 3.8',
         'Programming Language :: Python :: 3.9',
+        'Programming Language :: Python :: 3.10',
     ]
 )
diff --git a/python/setup_openvino.py b/python/setup_openvino.py
index b307965..5d8350b 100644
--- a/python/setup_openvino.py
+++ b/python/setup_openvino.py
@@ -54,7 +54,7 @@ setuptools.setup(
     include_package_data=True,
     install_requires=["pyclipper>=1.2.1", "openvino>=2022.2.0",
                       "opencv_python>=4.5.1.48", "numpy>=1.19.3",
-                      "six>=1.15.0", "Shapely>=1.7.1"],
+                      "six>=1.15.0", "Shapely>=1.7.1", 'PyYAML'],
     package_dir={'': MODULE_NAME},
     packages=setuptools.find_namespace_packages(where=MODULE_NAME),
     package_data={'': ['*.onnx', '*.yaml', '*.txt']},
@@ -66,5 +66,6 @@ setuptools.setup(
         'Programming Language :: Python :: 3.7',
         'Programming Language :: Python :: 3.8',
         'Programming Language :: Python :: 3.9',
+        'Programming Language :: Python :: 3.10',
     ]
 )