Fix

Signed-off-by: Joaquin Anton Guirao <[email protected]>
NVIDIA · Dec 23, 2024 · bdf5c5f · bdf5c5f
1 parent ce34260
commit bdf5c5f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dali/python/backend_impl.cc b/dali/python/backend_impl.cc
@@ -1179,7 +1179,7 @@ void ExposeTensorList(py::module &m) {
       Returns a `TensorListGPU` object being a copy of this `TensorListCPU`.
       )code",
       py::return_value_policy::take_ownership)
-    .def("as_cpu", [](TensorList<CPUBackend> &t) {
+    .def("as_cpu", [](TensorList<CPUBackend> &t) -> TensorList<CPUBackend> & {
         return t;
       }, R"code(Passthrough, as it is already an instance of `TensorListCPU`.)code",
       py::return_value_policy::reference_internal)