PY-50885 Fix CCE when resolving namespace package references in docstrings

GitOrigin-RevId: f8196bd41e5a0a880651fa08d3b6820594cf44b7

python/python-psi-impl/src/com/jetbrains/python/documentation/docstrings/DocStringTypeReference.java

@@ -77,13 +77,13 @@ public class DocStringTypeReference extends PsiPolyVariantReferenceBase<PsiEleme
 
   @Override
   public ResolveResult @NotNull [] multiResolve(boolean incompleteCode) {
-    PyElement result = null;
+    PsiElement result = null;
     final ResolveResultList results = new ResolveResultList();
     if (myType instanceof PyClassType) {
       result = ((PyClassType)myType).getPyClass();
     }
     else if (myType instanceof PyImportedModuleType) {
-      result = (PyElement)((PyImportedModuleType)myType).getImportedModule().resolve();
+      result = ((PyImportedModuleType)myType).getImportedModule().resolve();
     }
     else if (myType instanceof PyModuleType) {
       result = ((PyModuleType)myType).getModule();

python/testData/inspections/PyUnresolvedReferencesInspection3K/NamespacePackageReferenceInDocstringType/a.py

@@ -0,0 +1,7 @@
+import nspkg.mod
+
+
+def func(p):
+    """
+    :type p: nspkg.mod.Class
+    """

python/testData/inspections/PyUnresolvedReferencesInspection3K/NamespacePackageReferenceInDocstringType/nspkg/mod.py

@@ -0,0 +1,2 @@
+class Class:
+    pass

python/testSrc/com/jetbrains/python/inspections/Py3UnresolvedReferencesInspectionTest.java

@@ -311,4 +311,9 @@ public class Py3UnresolvedReferencesInspectionTest extends PyInspectionTestCase
   public void testClassLevelDunderAll() {
     doMultiFileTest("a.py");
   }
+
+  // PY-50885
+  public void testNamespacePackageReferenceInDocstringType() {
+    doMultiFileTest();
+  }
 }