MCPcopy Create free account
hub / github.com/SalesforceAIResearch/perfcodegen / visit_FunctionDef

Method visit_FunctionDef

src/sanitize.py:207–218  ·  view source on GitHub ↗
(self, node)

Source from the content-addressed store, hash-verified

205
206
207 def visit_FunctionDef(self, node):
208 #rename the first function generated as LLMs tend to generate extra useless code in the end of response
209 if not self.entry_point and self.mode == "funcname" and node.name == self.visitor.funcs[-1] and self.classname == "global":
210 self.ori_name = node.name
211 node.name = "solution"
212 elif self.entry_point and self.mode == "funcname" and node.name == self.entry_point and self.classname == "global":
213 self.ori_name = node.name
214 node.name = "solution"
215
216 self.generic_visit(node)
217
218 return node
219
220 def visit_Call(self, node):
221 if self.mode == "input" and isinstance(node.func, ast.Name) and node.func.id == "input":

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected