diff --git a/python/src/com/jetbrains/python/codeInsight/mlcompletion/PyCompletionFeatures.kt b/python/src/com/jetbrains/python/codeInsight/mlcompletion/PyCompletionFeatures.kt index 372549aed3f9..38044f24b9f2 100644 --- a/python/src/com/jetbrains/python/codeInsight/mlcompletion/PyCompletionFeatures.kt +++ b/python/src/com/jetbrains/python/codeInsight/mlcompletion/PyCompletionFeatures.kt @@ -164,9 +164,9 @@ object PyCompletionFeatures { fun getNumberOfOccurrencesInScope(kind: PyCompletionMlElementKind, contextFeatures: ContextFeatures, lookupString: String): Int? { return when (kind) { - in arrayOf(PyCompletionMlElementKind.FUNCTION, - PyCompletionMlElementKind.TYPE_OR_CLASS, - PyCompletionMlElementKind.FROM_TARGET) -> + PyCompletionMlElementKind.FUNCTION, + PyCompletionMlElementKind.TYPE_OR_CLASS, + PyCompletionMlElementKind.FROM_TARGET -> contextFeatures.getUserData(PyNamesMatchingMlCompletionFeatures.statementListOrFileNamesKey) ?.let { it[lookupString] } PyCompletionMlElementKind.NAMED_ARG ->