PY-4813 Performance tests fix

This commit is contained in:
Anton Bragin
2017-12-28 13:31:33 +03:00
parent 3d5f2d8508
commit ee405b15ca
@@ -282,7 +282,10 @@ public class PyTypingTypeProvider extends PyTypeProviderBase {
private static PyFunctionType getOverriddenFunctionType(@NotNull PyFunction function, @NotNull TypeEvalContext context) {
final PyFunction overriddenFunction = getOverriddenFunction(function, context);
if (overriddenFunction != null) {
return (PyFunctionType)context.getType(overriddenFunction);
PyType type = context.getType(overriddenFunction);
if (type instanceof PyFunctionType) {
return (PyFunctionType)context.getType(overriddenFunction);
}
}
return null;