PY-16987 Google and Numpy docstrings return null as parameter type if it wasn't speci...
[idea/community.git] / python / src / com / jetbrains / python / documentation / docstrings / SectionBasedDocString.java
index d0ba1cccbe74f1139f47bb78e7e707745a80a9c1..8b4743313b97a63cf10a737fe4796a2b3124b89c 100644 (file)
@@ -318,13 +318,8 @@ public abstract class SectionBasedDocString extends DocStringLineParser implemen
   @Nullable
   @Override
   public String getParamType(@Nullable String paramName) {
-    if (paramName != null) {
-      final SectionField field = getFirstFieldForParameter(paramName);
-      if (field != null) {
-        return field.getType();
-      }
-    }
-    return null;
+    final Substring sub = getParamTypeSubstring(paramName);
+    return sub != null ? sub.toString() : null;
   }
 
   @Nullable
@@ -429,8 +424,8 @@ public abstract class SectionBasedDocString extends DocStringLineParser implemen
   @Nullable
   @Override
   public String getReturnType() {
-    final SectionField field = getFirstReturnField();
-    return field != null ? field.getType() : null;
+    final Substring sub = getReturnTypeSubstring();
+    return sub != null ? sub.toString() : null;
   }
 
   @Nullable