diff --git a/mycodemod/replace_functions.py b/mycodemod/replace_functions.py index aee7b30..377d454 100644 --- a/mycodemod/replace_functions.py +++ b/mycodemod/replace_functions.py @@ -22,7 +22,7 @@ class ReplaceFunctionCommand(VisitorBasedCodemodCommand): def leave_FunctionDef(self, original_node: cst.FunctionDef, updated_node: cst.FunctionDef) -> cst.FunctionDef: functions_docstring = updated_node.get_docstring() - docstring_should_be = '' + docstring_should_be = '"""No docstring here yet."""' if functions_docstring is not None: docstring_should_be = '"""\n{}\n\n"""'.format(functions_docstring)