Upload folder using huggingface_hub

tasks.py

@@ -56,8 +56,11 @@ def _base_breakdown(action: Any, ground_truth: GroundTruth) -> dict:
     return breakdown
 
 
-def _reward_from_breakdown(breakdown: dict):
-    from env import Reward
+def _reward_from_breakdown(breakdown: dict):
+    try:
+        from .env import Reward
+    except ImportError:
+        from env import Reward
 
     total = round(sum(breakdown.values()), 4)
     return Reward(total=max(0.0, min(1.0, total)), breakdown=breakdown)
@@ -124,13 +127,16 @@ def confounded_hard_grader(trajectory: Any = None) -> float:
     return _delegate(trajectory)
 
 
-def _task_definition(
-    task_id: str,
-    difficulty: str,
-    description: str,
-    action_grader: Callable[[Any], Any],
-) -> TaskDefinition:
-    from env import AdverseEventReport
+def _task_definition(
+    task_id: str,
+    difficulty: str,
+    description: str,
+    action_grader: Callable[[Any], Any],
+) -> TaskDefinition:
+    try:
+        from .env import AdverseEventReport
+    except ImportError:
+        from env import AdverseEventReport
 
     task_data = TASK_DATA[task_id]
     return TaskDefinition(