fix: uses into_bound_py_any to cover more wider range

Jij-Inc · Jan 17, 2025 · cdf3e9e · cdf3e9e
1 parent 60f84a4
commit cdf3e9e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/pyo3-stub-gen-derive/src/gen_stub/signature.rs b/pyo3-stub-gen-derive/src/gen_stub/signature.rs
@@ -105,7 +105,7 @@ impl ToTokens for ArgsWithSignature<'_> {
                                         ::pyo3::prepare_freethreaded_python();
                                         ::pyo3::Python::with_gil(|py| -> String {
                                             let v: #ty = #value;
-                                            if let Ok(py_obj) = <#ty as ::pyo3::IntoPyObject>::into_pyobject(v, py) {
+                                            if let Ok(py_obj) = <#ty as ::pyo3::IntoPyObjectExt>::into_bound_py_any(v, py) {
                                                 ::pyo3_stub_gen::util::fmt_py_obj(&py_obj)
                                             } else {
                                                 "...".to_owned()

diff --git a/pyo3-stub-gen/src/util.rs b/pyo3-stub-gen/src/util.rs
@@ -1,7 +1,6 @@
 use pyo3::{prelude::*, types::*};
 
-pub fn all_builtin_types<T>(input: &Bound<'_, T>) -> bool {
-    let any = input.as_any();
+pub fn all_builtin_types(any: &Bound<'_, PyAny>) -> bool {
     if any.is_instance_of::<PyString>()
         || any.is_instance_of::<PyBool>()
         || any.is_instance_of::<PyInt>()
@@ -34,8 +33,7 @@ pub fn all_builtin_types<T>(input: &Bound<'_, T>) -> bool {
     false
 }
 
-pub fn fmt_py_obj<T>(input: &Bound<'_, T>) -> String {
-    let any = input.as_any();
+pub fn fmt_py_obj(any: &Bound<'_, PyAny>) -> String {
     if all_builtin_types(any) {
         if let Ok(py_str) = any.as_any().repr() {
             return py_str.to_string();