Gradio-Blocks/ML-Aided-Code-Analysis

1from src.utils import GenCode, Config2 3code_gen_class = GenCode(Config.code_gen_idt, gen_kwargs={})4 5def test_gen_code():6    code_snippet = "def dataset_name():\n    return 'test'"7    generated_code = code_gen_class(code_snippet)8    print(generated_code)9 10 11if __name__ == "__main__":12    test_gen_code()