@@ -5,7 +5,7 @@ import com.jetbrains.python.PyNames
55import com.jetbrains.python.PyTokenTypes
66import com.jetbrains.python.codeInsight.typing.PyTypingTypeProvider
77import com.jetbrains.python.psi.*
8- import com.jetbrains.python.psi.impl.PyCallExpressionHelper
8+ import com.jetbrains.python.psi.impl.PyCallExpressionNavigator
99import com.jetbrains.python.psi.types.*
1010import space.whitememory.pythoninlayparams.types.variables.PythonVariablesInlayTypeHintsProvider
1111
@@ -172,7 +172,7 @@ enum class HintResolver {
172172
173173 if (
174174 assignedValue is PyCallExpression
175- && PyCallExpressionHelper .resolveCalleeClass (assignedValue) != null
175+ && PyCallExpressionNavigator .getPyCallExpressionByCallee (assignedValue) != null
176176 ) {
177177 // Handle case like User().get_name() and list()
178178 if (typeAnnotation.isBuiltin || assignedValue.callee?.reference?.resolve() is PyFunction ) {
@@ -216,8 +216,8 @@ enum class HintResolver {
216216 }
217217
218218 if (expressionOperands.leftOperand is PyCallExpression && expressionOperands.rightOperand is PyCallExpression ) {
219- val isFalsePartClass = PyCallExpressionHelper .resolveCalleeClass (expressionOperands.leftOperand) != null
220- val isTruePartClass = PyCallExpressionHelper .resolveCalleeClass (expressionOperands.rightOperand) != null
219+ val isFalsePartClass = PyCallExpressionNavigator .getPyCallExpressionByCallee (expressionOperands.leftOperand) != null
220+ val isTruePartClass = PyCallExpressionNavigator .getPyCallExpressionByCallee (expressionOperands.rightOperand) != null
221221
222222 if (isFalsePartClass && isTruePartClass) return false
223223 }
@@ -264,9 +264,9 @@ enum class HintResolver {
264264
265265 if (assignmentValue !is PyCallExpression ) return true
266266
267- if (typeAnnotation is PyNoneType ) return true
267+ if (typeAnnotation is PyNoneLiteralExpression ) return true
268268
269- val resolvedClass = PyCallExpressionHelper .resolveCalleeClass (assignmentValue) ? : return true
269+ val resolvedClass = PyCallExpressionNavigator .getPyCallExpressionByCallee (assignmentValue) ? : return true
270270
271271 if (! collectionNames.contains(resolvedClass.name)) {
272272 return resolvedClass.name != typeAnnotation?.name
@@ -424,7 +424,7 @@ enum class HintResolver {
424424
425425 if (
426426 typeAnnotation == null
427- || (element is PyFunction && typeAnnotation is PyNoneType )
427+ || (element is PyFunction && typeAnnotation is PyNoneLiteralExpression )
428428 || ((element is PyFunction || element is PyTargetExpression ) && (element as PyTypeCommentOwner ).typeCommentAnnotation != null )
429429 || (element is PyAnnotationOwner && element.annotation != null )
430430 ) {
@@ -433,7 +433,7 @@ enum class HintResolver {
433433
434434 if (typeAnnotation is PyUnionType ) {
435435 return ! typeAnnotation.members.all {
436- PyTypeChecker .isUnknown(it, false , typeEvalContext) || (it is PyNoneType || it == null )
436+ PyTypeChecker .isUnknown(it, false , typeEvalContext) || (it is PyNoneLiteralExpression || it == null )
437437 }
438438 }
439439
0 commit comments