From deda03a5e9eab33371e80e3aca516493463fc280 Mon Sep 17 00:00:00 2001 From: Yu ISHIKAWA Date: Wed, 9 Sep 2015 16:30:04 +0900 Subject: [PATCH] [SPARK-10512][PySpark] Fix @since when a function doesn't have doc --- python/pyspark/__init__.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/python/pyspark/__init__.py b/python/pyspark/__init__.py index 8475dfb1c6ad0..afa866fcd2dad 100644 --- a/python/pyspark/__init__.py +++ b/python/pyspark/__init__.py @@ -57,9 +57,12 @@ def since(version): indent_p = re.compile(r'\n( +)') def deco(f): - indents = indent_p.findall(f.__doc__) - indent = ' ' * (min(len(m) for m in indents) if indents else 0) - f.__doc__ = f.__doc__.rstrip() + "\n\n%s.. versionadded:: %s" % (indent, version) + if f.__doc__ is None: + f.__doc__ = ".. versionadded:: %s" % (version) + else: + indents = indent_p.findall(f.__doc__) + indent = ' ' * (min(len(m) for m in indents) if indents else 0) + f.__doc__ = f.__doc__.rstrip() + "\n\n%s.. versionadded:: %s" % (indent, version) return f return deco