Function exec_safe

instruct_to_policy/scripts/src/utils.py:102–118

(code_str, gvars=None, lvars=None)

def exec_safe(code_str, gvars=None, lvars=None):
    # Due to few shot and zero shot prompt, code might generate import, slack this contraint
    # banned_phrases = ['import', '__']
    # for phrase in banned_phrases:
    #     assert phrase not in code_str
  
    if gvars is None:
        gvars = {}
    if lvars is None:
        lvars = {}
    empty_fn = lambda *args, **kwargs: None
    custom_gvars = merge_dicts([
        gvars,
        {'exec': empty_fn, 'eval': empty_fn}
    ])
    # print(code_str)
    exec(code_str, custom_gvars, lvars)