Python: minor bugfix in vector conversion

This commit is contained in:
Philippe Tillet
2015-12-22 16:43:55 -05:00
parent f2c3aa0acb
commit b322fe3942

View File

@@ -49,6 +49,8 @@ namespace tools
template<class T> template<class T>
std::vector<T> to_vector(bp::object const & iterable) std::vector<T> to_vector(bp::object const & iterable)
{ {
if(bp::extract<T>(iterable).check())
return {bp::extract<T>(iterable)};
std::size_t len = bp::len(iterable); std::size_t len = bp::len(iterable);
std::vector<T> res; res.reserve(len); std::vector<T> res; res.reserve(len);
for(std::size_t i = 0 ; i < len ; ++i) for(std::size_t i = 0 ; i < len ; ++i)