fix linting errors

aecay · aecay · commit 61496409faf9 · 2021-04-07T23:00:42.000+01:00
diff --git a/mypy/plugins/dataclasses.py b/mypy/plugins/dataclasses.py
@@ -13,7 +13,10 @@
     add_method, _get_decorator_bool_argument, deserialize_and_fixup_type,
 )
 from mypy.typeops import map_type_from_supertype
-from mypy.types import Type, Instance, NoneType, TypeVarDef, TypeVarType, CallableType, get_proper_type
+from mypy.types import (
+    Type, Instance, NoneType, TypeVarDef, TypeVarType, CallableType,
+    get_proper_type
+)
 from mypy.server.trigger import make_wildcard_trigger
 
 # The set of decorators that generate dataclasses.
@@ -365,7 +368,7 @@ def _propertize_callables(self, attributes: List[DataclassAttribute]) -> None:
         """
         info = self._ctx.cls.info
         for attr in attributes:
-            if isinstance(attr.type, CallableType):
+            if isinstance(get_proper_type(attr.type), CallableType):
                 var = attr.to_var()
                 var.info = info
                 var.is_property = True
