diff --git a/python/python-psi-impl/src/com/jetbrains/python/psi/impl/PyCallExpressionHelper.kt b/python/python-psi-impl/src/com/jetbrains/python/psi/impl/PyCallExpressionHelper.kt index 5d3f0ac14895..635e26e66481 100644 --- a/python/python-psi-impl/src/com/jetbrains/python/psi/impl/PyCallExpressionHelper.kt +++ b/python/python-psi-impl/src/com/jetbrains/python/psi/impl/PyCallExpressionHelper.kt @@ -127,9 +127,6 @@ fun PyCallExpression.getCalleeType(resolveContext: PyResolveContext): PyType? { return PyUnionType.union(callableTypes) } -fun multiResolveCallee(x: PyCallExpression, resolveContext: PyResolveContext): List = - x.multipleResolveCallee(resolveContext) - /** * It is not the same as [getCalleeType] since * this method returns callable types that would be actually called, the mentioned method returns type of underlying callee. @@ -141,13 +138,11 @@ fun multiResolveCallee(x: PyCallExpression, resolveContext: PyResolveContext): L * b = a() # callee type is A, resolved callee is A.__call__ * ``` */ -fun PyCallExpression.multipleResolveCallee(resolveContext: PyResolveContext): List { - return PyUtil.getParameterizedCachedValue( - this, - resolveContext) { - getExplicitResolveResults(it) + - getImplicitResolveResults(it) + - getRemoteResolveResults(it) +fun multiResolveCallee(expression: PyCallExpression, resolveContext: PyResolveContext): List { + return PyUtil.getParameterizedCachedValue(expression, resolveContext) { + expression.getExplicitResolveResults(it) + + expression.getImplicitResolveResults(it) + + expression.getRemoteResolveResults(it) } }