Merge branch 'weakref-call' of https://github.com/BobTheBuidler/mypy into weakref-call

Author: BobTheBuidler

mypyc/irbuild/specialize.py

@@ -31,7 +31,6 @@
     RefExpr,
     StrExpr,
     TupleExpr,
-    Var,
 )
 from mypy.types import AnyType, TypeOfAny
 from mypyc.ir.ops import (
@@ -145,10 +144,11 @@ def apply_function_specialization(
     """Invoke the Specializer callback for a function if one has been registered"""
     if (
         isinstance(callee, NameExpr)
-        and isinstance(callee.node, Var)
+        # and isinstance(callee.node, Var)
         # NOTE: why is this not a weakref rprimitive?
         # TODO: fix to weakref rprimitive so _apply_specialization can use the custom_op
-        and str(callee.node.type).startswith("weakref.ReferenceType")
+        and is_weakref_rprimitive(builder.node_type(callee.node))
+        # and str(callee.node.type).startswith("weakref.ReferenceType")
         and len(expr.args) == 0
     ):
         return builder.call_c(weakref_deref_op, [builder.accept(expr.callee)], expr.line)

mypyc/test-data/irbuild-weakref.test

@@ -69,4 +69,3 @@ L0:
     r = r0
     r1 = CPyWeakref_GetRef(r)
     return r1
-