Define capsule as Any type to fix pybind11-stubgen error.

isl-org · Aug 31, 2024 · 4d3fca4 · 4d3fca4
1 parent 2514991
commit 4d3fca4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cpp/pybind/core/tensor.cpp b/cpp/pybind/core/tensor.cpp
@@ -254,6 +254,7 @@ void pybind_core_tensor_declarations(py::module& m) {
     py::class_<Tensor> tensor(
             m, "Tensor",
             "A Tensor is a view of a data Blob with shape, stride, data_ptr.");
+    m.attr("capsule") = py::module_::import("typing").attr("Any");
 }
 void pybind_core_tensor_definitions(py::module& m) {
     auto tensor = static_cast<py::class_<Tensor>>(m.attr("Tensor"));