From f3fee4a4e1ac8a0ec2e545d0444b3a0342469aea Mon Sep 17 00:00:00 2001 From: Mark Aron Szulyovszky Date: Thu, 17 Feb 2022 16:53:09 +0100 Subject: [PATCH] fix(Codemod): restored original docstring replacement --- mycodemod/replace_functions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)