diff --git a/numpydoc/validate.py b/numpydoc/validate.py index b4b7e99f..d1ee978d 100644 --- 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: