Fix type check errors after updating mypy

ESSS · Nov 14, 2023 · fbcfc06 · fbcfc06
1 parent a1a0df9
commit fbcfc06
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/oop_ext/foundation/_tests/test_weak_ref.py b/src/oop_ext/foundation/_tests/test_weak_ref.py
@@ -318,7 +318,7 @@ def CustomAssertEqual(a, b):
         assert False, "{} != {}".format(a, b)
 
 
-def SetupTestAttributes():
+def SetupTestAttributes() -> Any:
     class C:
         x: int
 

diff --git a/src/oop_ext/foundation/callback/_callback.py b/src/oop_ext/foundation/callback/_callback.py
@@ -391,7 +391,7 @@ def Contains(
 
         real_func: Optional[Callable] = func
 
-        if real_func.__class__ == WeakMethodProxy:
+        if isinstance(real_func, WeakMethodProxy):
             real_func = cast(WeakMethodProxy, real_func)
             real_func = real_func.GetWrappedFunction()