diff options
Diffstat (limited to 'python/src/com/jetbrains/python/psi/impl/PyNamedParameterImpl.java')
-rw-r--r-- | python/src/com/jetbrains/python/psi/impl/PyNamedParameterImpl.java | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/python/src/com/jetbrains/python/psi/impl/PyNamedParameterImpl.java b/python/src/com/jetbrains/python/psi/impl/PyNamedParameterImpl.java index bf792fef0dae..cc9178c07f9e 100644 --- a/python/src/com/jetbrains/python/psi/impl/PyNamedParameterImpl.java +++ b/python/src/com/jetbrains/python/psi/impl/PyNamedParameterImpl.java @@ -181,11 +181,11 @@ public class PyNamedParameterImpl extends PyPresentableElementImpl<PyNamedParame PyParameterList parameterList = (PyParameterList)parent; PyFunction func = parameterList.getContainingFunction(); if (func != null) { - PyAnnotation anno = getAnnotation(); - if (anno != null) { - final PyClass pyClass = anno.resolveToClass(); - if (pyClass != null) { - return new PyClassTypeImpl(pyClass, false); + final PyAnnotation annotation = getAnnotation(); + if (annotation != null) { + final PyType type = context.getType(annotation); + if (type != null) { + return type; } } StructuredDocString docString = func.getStructuredDocString(); |