Update codebleu.py

codebleu.py

@@ -20,6 +20,7 @@ from .weighted_ngram_match import *
 from .syntax_match import *
 from .dataflow_match import *
 from tree_sitter import Language, Parser
+import os
 
 
 # TODO: Add BibTeX citation
@@ -138,7 +139,8 @@ class CodeBLEU(evaluate.Metric):
       # from os.path import isfile, join
       # onlyfiles = [f for f in listdir("./keywords") if isfile(join("keywords", f))]
       # print(onlyfiles)
-      keywords = [x.strip() for x in open('./keywords/'+ language +'.txt', 'r', encoding='utf-8').readlines()]
+      curr_path = os.path.dirname(os.path.abspath(__file__))
+      keywords = [x.strip() for x in open(curr_path + language +'.txt', 'r', encoding='utf-8').readlines()]
       def make_weights(reference_tokens, key_word_list):
           return {token:1 if token in key_word_list else 0.2 \
                   for token in reference_tokens}