Python: Add return type for eq/ne

mozilla · Jun 16, 2023 · 9daeb38 · 9daeb38
1 parent 3db8ec7
commit 9daeb38
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/uniffi_bindgen/src/bindings/python/templates/ObjectTemplate.py b/uniffi_bindgen/src/bindings/python/templates/ObjectTemplate.py
@@ -47,13 +47,13 @@ def {{ cons.name()|fn_name }}(cls, {% call py::arg_list_decl(cons) %}):
 {%-         when UniffiTrait::Display { fmt } %}
             {%- call py::method_decl("__str__", fmt) %}
 {%-         when UniffiTrait::Eq { eq, ne } %}
-    def __eq__(self, other: object):
+    def __eq__(self, other: object) -> {{ eq.return_type().unwrap()|type_name }}:
         if not isinstance(other, {{ type_name }}):
             return NotImplemented
 
         return {{ eq.return_type().unwrap()|lift_fn }}({% call py::to_ffi_call_with_prefix("self._pointer", eq) %})
 
-    def __ne__(self, other: object):
+    def __ne__(self, other: object) -> {{ ne.return_type().unwrap()|type_name }}:
         if not isinstance(other, {{ type_name }}):
             return NotImplemented