Fixed envs

src/custom_tasks/heq_task.py

@@ -9,7 +9,7 @@ from lighteval.tasks.requests import Doc
 from Levenshtein import distance
 import collections
 from lighteval.utils import as_list
-from envs import OWNER
+from ..envs import OWNER
 
 def get_tokens(s):
     if not s:

src/custom_tasks/sentiment_task.py

@@ -9,7 +9,7 @@ from lighteval.tasks.requests import Doc
 from Levenshtein import distance
 import collections
 from lighteval.utils import as_list
-from envs import OWNER
+from ..envs import OWNER
 
 def sentiment_eval_fn(golds: list[str], predictions: list[str], formatted_doc: Doc = None):
     if len(predictions)  > 1:

src/custom_tasks/translation_task.py

@@ -10,7 +10,7 @@ from Levenshtein import distance
 import collections
 from lighteval.utils import as_list
 import sacrebleu
-from envs import OWNER
+from ..envs import OWNER
 
 def trans_prompt_fn(line, task_name: str = None):
     """Defines how to go from a dataset line to a doc object.

src/custom_tasks/winograd_task.py

@@ -9,7 +9,7 @@ from lighteval.tasks.requests import Doc
 from Levenshtein import distance
 import collections
 from lighteval.utils import as_list
-from envs import OWNER
+from ..envs import OWNER
 
 def winograd_eval_fn(golds: list[str], predictions: list[str], formatted_doc: Doc = None):
     if len(predictions)  > 1: