buckification

Differential Revision: D43169568 Pull Request resolved: pytorch#692
kunalb · Feb 21, 2023 · 99ab917 · 99ab917
1 parent 9e2197b
commit 99ab917
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torchx/specs/file_linter.py b/torchx/specs/file_linter.py
@@ -75,7 +75,8 @@ def get_fn_docstring(fn: Callable[..., object]) -> Tuple[str, Dict[str, str]]:
         return default_fn_desc, args_description
     docstring = parse(func_description)
     for param in docstring.params:
-        args_description[param.arg_name] = param.description
+        if param.description is not None:
+            args_description[param.arg_name] = param.description
     short_func_description = docstring.short_description or default_fn_desc
     if docstring.long_description:
         short_func_description += " ..."