Use isgeneratorfunction to avoid overgenerating YD01 (#368)

jnothman · web-flow · commit 9b7ada8e8a89 · 2022-03-16T07:52:40.000+11:00
Fixes #365
diff --git a/numpydoc/validate.py b/numpydoc/validate.py
@@ -178,6 +178,10 @@ def type(self):
     def is_function_or_method(self):
         return inspect.isfunction(self.obj)
 
+    @property
+    def is_generator_function(self):
+        return inspect.isgeneratorfunction(self.obj)
+
     @property
     def source_file_name(self):
         """
@@ -596,7 +600,7 @@ def validate(obj_name):
             for name_or_type, type_, desc in doc.returns:
                 errs.extend(_check_desc(desc, "RT03", "RT04", "RT05"))
 
-        if not doc.yields and "yield" in doc.method_source:
+        if not doc.yields and doc.is_generator_function:
             errs.append(error("YD01"))
 
     if not doc.see_also:
