Update specialize.py

Author: BobTheBuidler

mypyc/irbuild/specialize.py

@@ -144,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)