From 76bd16748597a146b8afcd5e79521602d8ff201f Mon Sep 17 00:00:00 2001
From: Alejandro Saucedo <axsauze@gmail.com>
Date: Sat, 6 Mar 2021 08:05:09 +0000
Subject: [PATCH] Updated pthon to support desired extensions

---
 python/src/main.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/src/main.cpp b/python/src/main.cpp
index 594d9f45b..c851a2e43 100644
--- a/python/src/main.cpp
+++ b/python/src/main.cpp
@@ -135,8 +135,8 @@ PYBIND11_MODULE(kp, m) {
     py::class_<kp::Manager, std::shared_ptr<kp::Manager>>(m, "Manager")
         .def(py::init())
         .def(py::init<uint32_t>())
-        .def(py::init<uint32_t,const std::vector<uint32_t>&>())
-        .def("sequence", &kp::Manager::sequence, py::arg("queueIndex") = 0)
+        .def(py::init<uint32_t,const std::vector<uint32_t>&,const std::vector<std::string>&>())
+        .def("sequence", &kp::Manager::sequence, py::arg("queueIndex") = 0, py::arg("desired_extensions") = std::vector<std::string>())
         .def("tensor", [np](kp::Manager& self,
                             const py::array_t<float> data,
                             kp::Tensor::TensorTypes tensor_type) {