diff --git a/codescholar/apps/app_decl.py b/codescholar/apps/app_decl.py index ce629c9..13a745d 100644 --- a/codescholar/apps/app_decl.py +++ b/codescholar/apps/app_decl.py @@ -80,16 +80,13 @@ def search(): # search config args.mode = "q" args.seed = api - args.min_idiom_size = 2 + args.min_idiom_size = 3 args.max_idiom_size = 20 - args.max_init_beams = 150 + args.max_init_beams = 200 args.result_dir = f"{api_cache_dir}/{args.seed}/" args.idiom_g_dir = f"{args.result_dir}/idioms/graphs/" args.idiom_p_dir = f"{args.result_dir}/idioms/progs/" - if not osp.exists(args.idiom_g_dir): - os.makedirs(args.idiom_g_dir) - if not osp.exists(args.idiom_p_dir): os.makedirs(args.idiom_p_dir) diff --git a/codescholar/search/search.py b/codescholar/search/search.py index 7ac7da4..716d753 100644 --- a/codescholar/search/search.py +++ b/codescholar/search/search.py @@ -203,6 +203,9 @@ def main(args): else: prog_indices = grep_programs(args, args.seed)[: args.prog_samples] + if len(prog_indices) > 0: + return + # load all embeddings of prog_indices to redis # TODO: do this offline for *all* progs? load_embeddings_batched_redis(args, prog_indices)