Commitc471553

committed

Remove type hints from signature in solution

1 parent76dadf3 commitc471553Copy full SHA for c471553

File tree

-4

lines changed

-4

lines changed

Lines changed: 1 addition & 1 deletion

Lines changed: 7 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -31,9 +31,13 @@ def clean_program(program, *, inputs=None, function_name=None):`
`31`	`31`	`source=dedent(inspect.getsource(program))`
`32`	`32`	`iffunction_name:`
`33`	`33`	`assertsource.count("def solution(self, ")==1`
`34`		`-program=source.replace("def solution(self, ",f"def{function_name}(")`
`35`		`-# TODO strip annotations`
`36`		`-program=program.replace("@returns_stdout\n","")`
	`34`	`+params=", ".join(list(inspect.signature(program).parameters.keys())[1:])`
	`35`	`+program=re.sub(`
	`36`	`+"(@returns_stdout\n)?"`
	`37`	`+r"def solution\(self, .+?\):",`
	`38`	`+f"def{function_name}({params}):",`
	`39`	`+source,`
	`40`	`+ )`
`37`	`41`	`else:`
`38`	`42`	`atok=ASTTokens(source,parse=True)`
`39`	`43`	`func=atok.tree.body[0]`

Comments

(0)