Merge pull request #2 from Gamucopia-Creatives/feature/new-changes

envs/social_stream_moderation/environment.py

@@ -3,7 +3,7 @@ import os
 import random
 from typing import List, Dict, Any, Tuple, Optional
 from .models import HarmLabel, ModerationAction, State, PolicyMode, Post, UserGroup
-from .tasks import TASKS, TaskConfig
+from .tasks import TASKS, TaskConfig, resolve_task
 from .graders import compute_per_post_reward, grade_episode, get_grader
 
 class SocialStreamModerationEnv:
@@ -30,10 +30,10 @@ class SocialStreamModerationEnv:
         if seed is not None:
             random.seed(seed)
             
-        if task_name not in TASKS:
+        try:
+            self.current_task = resolve_task(task_name)
+        except KeyError:
             raise ValueError(f"Task {task_name} not found in TASKS.")
-            
-        self.current_task = TASKS[task_name]
         data_path = os.path.join(self.data_dir, self.current_task.data_file)
         
         with open(data_path, "r") as f:

envs/social_stream_moderation/tasks.py

@@ -40,3 +40,22 @@ TASKS = {
         grader_id="fairness_bias_grader"
     )
 }
+
+# Legacy aliases used by validate_submission.py and inference.py CLI.
+# Kept in a separate dict so that TASKS.values() only yields the 3 canonical
+# entries (avoids duplicates in the /tasks API endpoint).
+TASK_ALIASES = {
+    "clear_cut_moderation": "Task 1: Basic Safety",
+    "nuanced_sarcastic": "Task 2: Context & Nuance",
+    "policy_fairness": "Task 3: Fairness & Bias",
+}
+
+
+def resolve_task(name: str) -> TaskConfig:
+    """Look up a task by canonical name or legacy alias."""
+    if name in TASKS:
+        return TASKS[name]
+    canonical = TASK_ALIASES.get(name)
+    if canonical and canonical in TASKS:
+        return TASKS[canonical]
+    raise KeyError(f"Unknown task: {name}")

openenv.yaml

@@ -24,6 +24,19 @@ tasks:
     difficulty: hard
     description: "Ensure fairness across user groups and adhere to stricter policy regimes."
     grader_id: fairness_bias_grader
+  # Legacy aliases used by validate_submission.py
+  - id: clear_cut_moderation
+    difficulty: easy
+    description: "Alias for Task 1: Basic Safety"
+    grader_id: basic_safety_grader
+  - id: nuanced_sarcastic
+    difficulty: medium
+    description: "Alias for Task 2: Context & Nuance"
+    grader_id: context_nuance_grader
+  - id: policy_fairness
+    difficulty: hard
+    description: "Alias for Task 3: Fairness & Bias"
+    grader_id: fairness_bias_grader
 
 schemas:
   state: