filter languages

VHellendoorn · Mar 3, 2022 · 504aafb · 504aafb
1 parent a5ec32b
commit 504aafb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Evaluation/eval_codex_all.py b/Evaluation/eval_codex_all.py
@@ -62,7 +62,7 @@ def call_codex(code_str, save_probs):
     excluded_dirs = args.dirs + '-excluded'
     pathlib.Path(excluded_dirs).mkdir(parents=True, exist_ok=True)
     for language in dirs:
-        if language not in languages_to_run:
+        if language.split('/')[-1] not in languages_to_run:
             continue
         print('Language:', language)
         files = glob.glob(os.path.join(language, '**/*'), recursive=True)