Explicitly set the return type to match the python bindings

nv-morpheus · Sep 16, 2024 · e9ea972 · e9ea972
1 parent 831925a
commit e9ea972
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/mrc/_pymrc/src/segment.cpp b/python/mrc/_pymrc/src/segment.cpp
@@ -277,9 +277,9 @@ class SubscriberFuncWrapper : public mrc::pymrc::PythonSource<PyHolder>
             {
                 DVLOG(10) << ctx.info() << " Starting source";
                 py::gil_scoped_acquire gil;
-                auto subscription = subscriber.get_subscription();
-                py::object py_sub = py::cast(subscription);
-                auto py_iter      = m_gen_factory.operator()<py::iterator>(std::move(py_sub));
+                PySubscription subscription = subscriber.get_subscription();
+                py::object py_sub           = py::cast(subscription);
+                auto py_iter                = m_gen_factory.operator()<py::iterator>(std::move(py_sub));
                 PyIteratorWrapper iter_wrapper{std::move(py_iter)};
 
                 for (auto next_val : iter_wrapper)