Use isgeneratorfunction to avoid overgenerating YD01jnothman-patch-1

Fixes #365

Could do with a test

1 files changed, 5 insertions, 1 deletions

diff --git a/numpydoc/validate.py b/numpydoc/validate.py
index b4b7e99..d1ee978 100644
--- a/numpydoc/validate.py
+++ b/numpydoc/validate.py
@@ -179,6 +179,10 @@ class Validator:
         return inspect.isfunction(self.obj)
 
     @property
+    def is_generator_function(self):
+        return inspect.isgeneratorfunction(self.obj)
+
+    @property
     def source_file_name(self):
         """
         File name where the object is implemented (e.g. pandas/core/frame.py).
@@ -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: