a b2@s8ddlmZmZeejidZddZedkr4edS))GenCodeConfig)Z gen_kwargscCsd}t|}t|dS)Nz%def dataset_name(): return 'test')code_gen_classprint)Z code_snippetZgenerated_coders/Users/reshinthadithyan/master/research/code-research/weekend-hacks/ml-aided-static-code-analysis/src/test_utils.py test_gen_codesr__main__N)Z src.utilsrrZ code_gen_idtrr__name__rrrrs