cogames 0.3.64__py3-none-any.whl → 0.3.65__py3-none-any.whl

cogames/core.py

@@ -0,0 +1,87 @@
+"""Core base classes for CoGame missions and variants."""
+
+from __future__ import annotations
+
+from abc import ABC
+from typing import TypeVar
+
+from pydantic import Field
+from typing_extensions import Self
+
+from mettagrid.base_config import Config
+from mettagrid.config.mettagrid_config import MettaGridConfig
+from mettagrid.map_builder.map_builder import AnyMapBuilderConfig
+
+# Type variable for mission types
+TMission = TypeVar("TMission", bound="CoGameMission")
+
+MAP_MISSION_DELIMITER = "."
+
+
+class CoGameMissionVariant(Config, ABC):
+    # Note: we could derive the name from the class name automatically, but it would make it
+    # harder to find the variant source code based on CLI interactions.
+    name: str
+    description: str = Field(default="")
+
+    def modify_mission(self, mission: CoGameMission) -> None:
+        # Override this method to modify the mission.
+        # Variants are allowed to modify the mission in-place - it's guaranteed to be a one-time only instance.
+        pass
+
+    def modify_env(self, mission: CoGameMission, env: MettaGridConfig) -> None:
+        # Override this method to modify the produced environment.
+        # Variants are allowed to modify the environment in-place.
+        pass
+
+    def compat(self, mission: CoGameMission) -> bool:
+        """Check if this variant is compatible with the given mission.
+
+        Returns True if the variant can be safely applied to the mission.
+        Override this method to add compatibility checks.
+        """
+        return True
+
+    def apply(self, mission: TMission) -> TMission:
+        mission = mission.model_copy(deep=True)
+        mission.variants.append(self)
+        self.modify_mission(mission)
+        return mission
+
+
+class CoGameSite(Config):
+    name: str
+    description: str
+    map_builder: AnyMapBuilderConfig
+
+    min_cogs: int = Field(default=1, ge=1)
+    max_cogs: int = Field(default=1000, ge=1)
+
+
+class CoGameMission(Config, ABC):
+    """Base class for Mission configurations with common fields and methods."""
+
+    name: str
+    description: str
+    site: CoGameSite
+    num_cogs: int | None = None
+
+    # Variants are applied to the mission immediately, and to its env when make_env is called
+    variants: list[CoGameMissionVariant] = Field(default_factory=list)
+
+    max_steps: int = Field(default=10000)
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        # Can't call `variant.apply` here because it will create a new mission instance
+        for variant in self.variants:
+            variant.modify_mission(self)
+
+    def with_variants(self, variants: list[CoGameMissionVariant]) -> Self:
+        mission = self
+        for variant in variants:
+            mission = variant.apply(mission)
+        return mission
+
+    def full_name(self) -> str:
+        return f"{self.site.name}{MAP_MISSION_DELIMITER}{self.name}"

cogames/verbose.py

@@ -4,14 +4,14 @@ import json
 
 from rich.console import Console
 
-from cogames.cogs_vs_clips.mission import AnyMission
+from cogames.cogs_vs_clips.mission import CvCMission
 from mettagrid.config.mettagrid_config import MettaGridConfig
 
 
 def print_configs(
     console: Console,
     env_cfg: MettaGridConfig,
-    mission_cfg: AnyMission | None = None,
+    mission_cfg: CvCMission | None = None,
     print_cvc_config: bool = False,
     print_mg_config: bool = False,
 ) -> None:

{cogames-0.3.64.dist-info → cogames-0.3.65.dist-info}/METADATA

@@ -1,13 +1,13 @@
 Metadata-Version: 2.4
 Name: cogames
-Version: 0.3.64
+Version: 0.3.65
 Summary: Multi-agent cooperative games
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Requires-Python: <3.13,>=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: mettagrid==0.2.0.74
+Requires-Dist: mettagrid==0.2.0.75
 Requires-Dist: packaging>=24.0.0
 Requires-Dist: pufferlib-core
 Requires-Dist: pydantic>=2.11.5

{cogames-0.3.64.dist-info → cogames-0.3.65.dist-info}/RECORD

@@ -1,5 +1,6 @@
 cogames/__init__.py,sha256=Loao4kW7AqCRVKcQuA7lKEsyKl9k_Zf1oVlPXvgqjbw,23
 cogames/auth.py,sha256=W7BgRhrCTKob68p6HHTKVFjPayySaBG03eIgcGjw_jI,15852
+cogames/core.py,sha256=ISoWiRQXuqJhMyN-tROvmIU0_c9s5H__pJVE8BLQCyw,2859
 cogames/curricula.py,sha256=a7Nd-av4epAjmoqlefyU4q5eAitpYjCEGDCF_tS_iaY,825
 cogames/device.py,sha256=GVC7g4tNVySn_rSbHJB0jGKvpYBL8-VmeYEQXWXtvy0,1680
 cogames/evaluate.py,sha256=xm4dCLAXZFhfygDeC_617QXRa5DQYB_6EENEeeDAiWQ,9287
@@ -9,14 +10,14 @@ cogames/pickup.py,sha256=gCm-BLMSKN4iflyn2EdsxE7ywKYzMgjozep5o_306yE,6545
 cogames/play.py,sha256=SV_gKg5kNj68m7bavmEYomLmmWxVGbc50ZhwAEMDy5E,8014
 cogames/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cogames/train.py,sha256=C4ZP00fxHspiQK-p00EPXyfcwiFKNIBUF1iBe2EE0GE,17894
-cogames/verbose.py,sha256=WcbVMpB43YpNJxZcwEkFtIqiIIxPLMZMu_3XQmeY59w,2916
+cogames/verbose.py,sha256=cqdGgSjoTQFektxfDglEZgobm1zfWsxReG2jLborzqI,2916
 cogames/cli/base.py,sha256=UpMqeJ0C8i0pkue1AHnnyyX0bFJ9zZeJ7HBR6yhuA8A,54
 cogames/cli/client.py,sha256=UFWn6Kjbp6nCYjeC5wMwIJAM5o_hngNdZ_yv24aeiOI,9025
 cogames/cli/leaderboard.py,sha256=eIbiSBOkvoi5IL2weiKBOJAjWGHquDk1PWoHKA8uu3g,13941
 cogames/cli/login.py,sha256=_i1Hdbp_wAMsX0NLbCTSr7GmOFbzSqwyNP79OqLOg40,1237
-cogames/cli/mission.py,sha256=Shd7wivVdRbWTZ02yH334QZfwAqNy6mvRzLvU01iJrA,21674
+cogames/cli/mission.py,sha256=GyHDSDjLnQ0zRp-zk5kOO90IUZE9_860uAfCb_4L1v0,21354
 cogames/cli/policy.py,sha256=JI_aX0lcY-kaZWl-RhUfFu0vknF2mXHS8y0CQRRhpao,8743
-cogames/cli/submit.py,sha256=vB8EhB1z-SOUIdRhIkMEm_sbjFXbarVUn5BuiPa7SMI,20927
+cogames/cli/submit.py,sha256=jH0ZLdypL43UkrqXmix1-QwwJhv0G_WVVqsiLNaJ1n0,20897
 cogames/cli/utils.py,sha256=K4eb-RL73HtavNGxZjn_SVDLAwfFamLCEQNTCNVdmPU,895
 cogames/cli/docsync/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cogames/cli/docsync/_nb_md_directive_processing.py,sha256=A920oS21mw9ZU2vp-cILzq5Sa-AMtmwMSayZKcQwX9M,6365
@@ -26,24 +27,24 @@ cogames/cli/docsync/_three_way_sync.py,sha256=fug-HXNYqJhJvusWHnGmLQewtiAUKhCBOd
 cogames/cli/docsync/_utils.py,sha256=zPfXzVa0XbqZReGvMUvxl4yLWc2oFeJJiefOBjOHaCA,2616
 cogames/cli/docsync/docsync.py,sha256=dtGG91Co89VdiwRwKk57kYd-qnHiFIgIkS1OI12AYyQ,5664
 cogames/cogs_vs_clips/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cogames/cogs_vs_clips/cog.py,sha256=P6T9y6iZ4aAXATqmQ3ooRLvP5K60r3o-viX4eoiaZq4,3201
-cogames/cogs_vs_clips/cogs_vs_clips_mapgen.md,sha256=Tof8NFILYAj4nTYXFXgbGlLaryHj0Zvjdm-lBGKP5dE,13378
-cogames/cogs_vs_clips/cogsguard_reward_variants.py,sha256=V54Ly7T84cRfBoHv7Am7tv2OpL43Ksm95Fqkwy_TWAk,5819
-cogames/cogs_vs_clips/cogsguard_tutorial.py,sha256=_J4yCpf4783Cp4Z0dV6aRY0DCWDxQ4P4PBVhePRMQIw,1623
-cogames/cogs_vs_clips/mission.py,sha256=KTpfBPiwzAxobmnQ-WqPn0dLiAR4goFeHUAeyO4ahgA,12480
-cogames/cogs_vs_clips/mission_utils.py,sha256=-tB8JclTaq8Zfh8V5DPor1A3Ncuq8ndsTjgVnIxUV7s,671
-cogames/cogs_vs_clips/missions.py,sha256=ZAmS7qYViKwF4XazyqE7BE5HEhgBhwWhWCCYBfUCU7Y,2430
-cogames/cogs_vs_clips/procedural.py,sha256=M_a6uRsHzatVSW8PrVTAupv2u1M_ehOgM6XnEB3_Gbk,25947
-cogames/cogs_vs_clips/sites.py,sha256=Mq10ammCu49-xngQtJy8UwfRR6d78FN2dzxxi7HZy04,4916
-cogames/cogs_vs_clips/stations.py,sha256=HxTPQnY2Y8kqK42OnaSAg8S69yO7g2pr-WkGiaL967I,9174
-cogames/cogs_vs_clips/tutorial_missions.py,sha256=p2ds5t_cfD9xyFLSq8A9xw-zO4-OHi9LSGdocl4JDC4,712
-cogames/cogs_vs_clips/variants.py,sha256=zU32g8Hy2GEBgOMjf6VU6Phn4X7wVvfV4_shjtddOc0,11783
-cogames/cogs_vs_clips/evals/README.md,sha256=aglXp04_1AXwxk-zfYsqOi3qbVpSRvOVJY3j5I_yCSA,7048
-cogames/cogs_vs_clips/evals/cogsguard_evals.py,sha256=OgIFjQIUbEbjyWxalr4_eRPYoi3bkgvFNsZfW9kKu7I,2510
-cogames/cogs_vs_clips/evals/diagnostic_evals.py,sha256=RakNmoy8I5lgYqlUT38sctIszaLWLQmnekaNjg---Ck,13626
-cogames/cogs_vs_clips/evals/difficulty_variants.py,sha256=SCUAbbOCCYuK2gEcuJMWA-qAlQfme9AjEOIESFj6DBY,4319
-cogames/cogs_vs_clips/evals/integrated_evals.py,sha256=wTq2JOeF4pxU59TFn0VLs7atNiVG4VFfmF-z8sFuERY,2784
-cogames/cogs_vs_clips/evals/spanning_evals.py,sha256=CWa0HtG6LkxVP-SDDeWCSOC9Tw_20fFoAtQurZHJzTo,7516
+cogames/cogs_vs_clips/clips.py,sha256=FlaG2GNx605eSgf3dU2G5GbgtgnG0rAIrhsdjuvKkKs,3611
+cogames/cogs_vs_clips/cog.py,sha256=rcFwpG24MHjHx6Erk9sj-bHTvAKgSfExB0YZbFnRwvk,3596
+cogames/cogs_vs_clips/cogsguard_tutorial.py,sha256=FjVep1F2q2W9pieCbbJZIR21ImT4ylMiR87P9r98zFI,1577
+cogames/cogs_vs_clips/config.py,sha256=TbDGpU_tS0ClG92uvBNzWh-VzFPMko2tt8r_QYj5R7U,1178
+cogames/cogs_vs_clips/mission.py,sha256=NRidQF81g0pVcZzGnlTpjXZRcMuojP0-tNMwdJdSzR0,4413
+cogames/cogs_vs_clips/missions.py,sha256=feOLCwkpEIcBP6-HeKlo_AyP0-eZmdV58LdG-4gkTLw,2116
+cogames/cogs_vs_clips/sites.py,sha256=2GTKwJweipbKVKIameMZYEO2hTixBoEtrVpyjSGHGX8,5446
+cogames/cogs_vs_clips/stations.py,sha256=pZCA2_AAcAhV0QjLpFg1rqgLS9LEsID60lTGcMNARw8,7800
+cogames/cogs_vs_clips/team.py,sha256=o3FceRfRqiOdwKkO8SGAMw4kCnEOPE9OGrjm_ucZYN8,1480
+cogames/cogs_vs_clips/terrain.py,sha256=TeKIydkdZ4prrPfRPy8pE-_5OiNSzh2g5Pjt31gZA40,26105
+cogames/cogs_vs_clips/variants.py,sha256=0AuwIx9IQjxMSRtp5OUeQw9qDqCtdAjRpmAQGkzBnoE,7899
+cogames/cogs_vs_clips/docs/cogs_vs_clips_mapgen.md,sha256=5vgdYPKWryeh6qEV8nY1bKyDdqriQXGwh8JuGnSoQqE,13198
+cogames/cogs_vs_clips/evals/README.md,sha256=sHlK2XDOPK0L3kphSvYWeKqVUjEg9p6TjAa8yhMtKlw,7053
+cogames/cogs_vs_clips/evals/cogsguard_evals.py,sha256=LiV6J8WIoUf6pWoWN1DcP9UdyNo7XHGzhhuraw4ZNQ8,2952
+cogames/cogs_vs_clips/evals/diagnostic_evals.py,sha256=29sIV_sOC3XWA7s77_dPE0MEbFhqFYQ71rmckx5e9iU,10123
+cogames/cogs_vs_clips/evals/difficulty_variants.py,sha256=tm5cFdOzE_HoaOKZ_Mqb0I1vnmNOEpWfjZoYTRGz-14,3803
+cogames/cogs_vs_clips/evals/integrated_evals.py,sha256=SfR-VTzEjLIViqpbeCnVHDqQMocK6cBstXD7CshdI6g,1433
+cogames/cogs_vs_clips/evals/spanning_evals.py,sha256=PtHABzA_rlM_uyqo-RTLl3lCC84GZZdjrnPevRfT1Ts,7340
 cogames/docs/SCRIPTED_AGENT.md,sha256=MDvYQ1Xccexnr_HjCnF5hNsZtqGAy2s0VA3tdgZHego,10918
 cogames/maps/canidate1_1000.map,sha256=DjLOgWDX9e9bmnA7oXcyIR-yilpoWufW0tsU12wSMmI,1023273
 cogames/maps/canidate1_1000_stations.map,sha256=RXqSceZiEf6TmEfeFO8vsNiAnyhox4zCLp77xPwSO7g,1023402
@@ -144,7 +145,7 @@ cogames/policy/trainable_policy_template.py,sha256=GbB4vi0wsIiuh_APtfjnavvh9ZVUx
 cogames/policy/nim_agents/__init__.py,sha256=q22rmDMKHMO2A4Cqt0P-kuH3f6yaldlaBaRuGydeEKM,1247
 cogames/policy/nim_agents/agents.py,sha256=2uxzs8yxcz3obdA_gaWf0qgojFbXSIU0W3cS-9XD0AI,5038
 cogames/policy/nim_agents/thinky_eval.py,sha256=NqIZng30MpyZD5hF03uoje4potd873hWyHOb3AZdeZk,1154
-cogames-0.3.64.dist-info/licenses/LICENSE,sha256=NG4hf0NHdGZhkabVCK1MKX8RAJmWaEm4eaGcUsONZ_E,1065
+cogames-0.3.65.dist-info/licenses/LICENSE,sha256=NG4hf0NHdGZhkabVCK1MKX8RAJmWaEm4eaGcUsONZ_E,1065
 metta_alo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 metta_alo/job_specs.py,sha256=_QT23tGb7V8W1jqiUijeQUQh6ZvKfW5-KYiYSuNTdVI,450
 metta_alo/policy.py,sha256=RnBwDzQ7abFiQSJe6MYr_JtsyHlKBES1CkDN6mZuvQI,613
@@ -152,8 +153,8 @@ metta_alo/pure_single_episode_runner.py,sha256=VsSBEn2BVWWHtiZ0SdL9_QMTom_vbbIyP
 metta_alo/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 metta_alo/rollout.py,sha256=hc7JRQD4ilfqTO57BvKav2v_JW1Vbcc-oYcEmoWVlB4,10192
 metta_alo/scoring.py,sha256=7ikuP6Vnvv57bAc_ARbTadDSf6wm5dCY49xX9QjD_fs,5809
-cogames-0.3.64.dist-info/METADATA,sha256=Vnhu2_9kNcq7ALn4VF4VFMP8j25Mj6ZE1f-d1p7zBU4,234991
-cogames-0.3.64.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cogames-0.3.64.dist-info/entry_points.txt,sha256=GTHdgj-RC2hQYmjUMSp9RHX8kbru19k0LS2lAj8DnLE,45
-cogames-0.3.64.dist-info/top_level.txt,sha256=YErBkYWJd3-eksLpbgbMrETni1MPBNL4mqEyhZUa0UE,18
-cogames-0.3.64.dist-info/RECORD,,
+cogames-0.3.65.dist-info/METADATA,sha256=MmW7r978aI0QKGZoD7J47oVHGGz9mmIu8P26htS2ZCY,234991
+cogames-0.3.65.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cogames-0.3.65.dist-info/entry_points.txt,sha256=GTHdgj-RC2hQYmjUMSp9RHX8kbru19k0LS2lAj8DnLE,45
+cogames-0.3.65.dist-info/top_level.txt,sha256=YErBkYWJd3-eksLpbgbMrETni1MPBNL4mqEyhZUa0UE,18
+cogames-0.3.65.dist-info/RECORD,,

cogames/cogs_vs_clips/cogsguard_reward_variants.py

@@ -1,153 +0,0 @@
-"""Reward preset wiring for the CogsGuard (Cogs vs Clips) mission.
-
-The mission has a single "true" objective signal, plus optional shaping variants.
-Reward variants are stackable; each one adds additional shaping signals on top of the
-mission's default objective rewards.
-"""
-
-from __future__ import annotations
-
-from typing import Literal, Sequence, cast
-
-from mettagrid.config.game_value import stat
-from mettagrid.config.mettagrid_config import MettaGridConfig
-from mettagrid.config.reward_config import AgentReward, reward
-
-CogsGuardRewardVariant = Literal["credit", "milestones", "no_objective", "penalize_vibe_change", "objective"]
-
-AVAILABLE_REWARD_VARIANTS: tuple[CogsGuardRewardVariant, ...] = (
-    "objective",
-    "no_objective",
-    "milestones",
-    "credit",
-    "penalize_vibe_change",
-)
-
-_OBJECTIVE_STAT_KEY = "aligned_junction_held"
-
-
-def _apply_milestones(rewards: dict[str, AgentReward], *, max_junctions: int = 100) -> None:
-    """Add milestone shaping rewards onto an existing baseline.
-
-    Args:
-        rewards: Rewards dict to modify in-place.
-        max_junctions: Maximum expected number of junctions for capping rewards.
-            Defaults to 100 as a reasonable upper bound for most maps.
-    """
-    w_junction_aligned = 1.0
-    w_scramble_act = 0.5
-    w_align_act = 1.0
-
-    # Max caps based on expected junction counts
-    max_junction_aligned = w_junction_aligned * max_junctions
-    max_scramble = w_scramble_act * max_junctions
-    max_align = w_align_act * max_junctions
-
-    rewards["aligned_junctions"] = reward(
-        stat("collective.junction"),
-        weight=w_junction_aligned,
-        max=max_junction_aligned,
-    )
-
-    rewards["junction_scrambled_by_agent"] = reward(
-        stat("junction.scrambled_by_agent"),
-        weight=w_scramble_act,
-        max=max_scramble,
-    )
-    rewards["junction_aligned_by_agent"] = reward(
-        stat("junction.aligned_by_agent"),
-        weight=w_align_act,
-        max=max_align,
-    )
-
-
-def _apply_penalize_vibe_change(rewards: dict[str, AgentReward]) -> None:
-    """Add penalty for vibe changes to discourage spamming."""
-    w_vibe_change = -0.01
-    rewards["vibe_change_penalty"] = reward(stat("action.change_vibe.success"), weight=w_vibe_change)
-
-
-def _apply_credit(rewards: dict[str, AgentReward]) -> None:
-    """Add dense precursor shaping rewards onto an existing baseline."""
-    w_heart = 0.05
-    cap_heart = 0.5
-    w_align_gear = 0.2
-    cap_align_gear = 0.4
-    w_scramble_gear = 0.2
-    cap_scramble_gear = 0.4
-    w_element_gain = 0.001
-    cap_element_gain = 0.1
-
-    # Stats rewards for gains as a single map
-    gain_rewards: dict[str, AgentReward] = {
-        "heart_gained": reward(stat("heart.gained"), weight=w_heart, max=cap_heart),
-        "aligner_gained": reward(stat("aligner.gained"), weight=w_align_gear, max=cap_align_gear),
-        "scrambler_gained": reward(stat("scrambler.gained"), weight=w_scramble_gear, max=cap_scramble_gear),
-        "carbon_gained": reward(stat("carbon.gained"), weight=w_element_gain, max=cap_element_gain),
-        "oxygen_gained": reward(stat("oxygen.gained"), weight=w_element_gain, max=cap_element_gain),
-        "germanium_gained": reward(stat("germanium.gained"), weight=w_element_gain, max=cap_element_gain),
-        "silicon_gained": reward(stat("silicon.gained"), weight=w_element_gain, max=cap_element_gain),
-    }
-    rewards.update(gain_rewards)
-
-    # Collective deposit rewards
-    w_deposit = 0.002
-    cap_deposit = 0.2
-    deposit_rewards: dict[str, AgentReward] = {
-        f"collective_{element}_deposited": reward(
-            stat(f"collective.{element}.deposited"), weight=w_deposit, max=cap_deposit
-        )
-        for element in ["carbon", "oxygen", "germanium", "silicon"]
-    }
-    rewards.update(deposit_rewards)
-
-
-def apply_reward_variants(env: MettaGridConfig, *, variants: str | Sequence[str] | None = None) -> None:
-    """Apply CogsGuard reward variants to `env`.
-
-    Variants are stackable:
-    - `objective`: no-op marker; keeps the mission's default objective reward wiring.
-    - `no_objective`: disables the objective stat reward (`junction.held`).
-    - `milestones`: adds shaped rewards for aligning/scrambling junctions and holding more junctions.
-    - `credit`: adds additional dense shaping for precursor behaviors (resources/gear/deposits).
-    - `penalize_vibe_change`: adds a penalty for vibe changes to discourage spamming.
-    """
-    if not variants:
-        return
-
-    variant_names = [variants] if isinstance(variants, str) else list(variants)
-
-    reward_variants: list[CogsGuardRewardVariant] = []
-    for variant_name in variant_names:
-        if variant_name not in AVAILABLE_REWARD_VARIANTS:
-            available = ", ".join(AVAILABLE_REWARD_VARIANTS)
-            raise ValueError(f"Unknown Cogsguard reward variant '{variant_name}'. Available: {available}")
-        variant = cast(CogsGuardRewardVariant, variant_name)
-        if variant in reward_variants:
-            continue
-        reward_variants.append(variant)
-
-    enabled = set(reward_variants)
-    if enabled <= {"objective"}:
-        return
-
-    # Start from the mission's existing objective baseline to preserve its scaling.
-    rewards = dict(env.game.agent.rewards)
-
-    if "no_objective" in enabled:
-        rewards.pop(_OBJECTIVE_STAT_KEY, None)
-    if "milestones" in enabled:
-        _apply_milestones(rewards)
-    if "credit" in enabled:
-        _apply_credit(rewards)
-    if "penalize_vibe_change" in enabled:
-        _apply_penalize_vibe_change(rewards)
-
-    env.game.agent.rewards = rewards
-
-    # Deterministic label suffix order (exclude "objective").
-    for variant in AVAILABLE_REWARD_VARIANTS:
-        if variant == "objective":
-            continue
-        if variant in enabled:
-            env.label += f".{variant}"

cogames/cogs_vs_clips/mission_utils.py

@@ -1,19 +0,0 @@
-"""Utility functions for mission configuration."""
-
-from __future__ import annotations
-
-from pathlib import Path
-
-from mettagrid.map_builder.map_builder import MapBuilderConfig
-from mettagrid.mapgen.mapgen import MapGen
-
-
-def get_map(map_name: str, fixed_spawn_order: bool = False) -> MapBuilderConfig:
-    """Load a map by name from the maps directory."""
-    maps_dir = Path(__file__).parent.parent / "maps"
-    map_path = maps_dir / map_name
-    return MapGen.Config(
-        instance=MapBuilderConfig.from_uri(str(map_path)),
-        fixed_spawn_order=fixed_spawn_order,
-        instance_border_width=0,  # Don't add border - maps already have borders built in
-    )

cogames/cogs_vs_clips/tutorial_missions.py

@@ -1,25 +0,0 @@
-from typing import override
-
-from cogames.cogs_vs_clips.mission import Mission, MissionVariant
-from cogames.cogs_vs_clips.sites import TRAINING_FACILITY
-
-
-class TutorialVariant(MissionVariant):
-    name: str = "tutorial_mode"
-    description: str = "High energy regen for learning."
-
-    @override
-    def modify_mission(self, mission: Mission) -> None:
-        mission.cog.energy_regen = 1
-
-    @override
-    def modify_env(self, mission: Mission, env) -> None:
-        env.game.max_steps = max(env.game.max_steps, 1000)
-
-
-TutorialMission = Mission(
-    name="tutorial",
-    description="Learn the basics of CoGames: Gather, Craft, and Deposit.",
-    site=TRAINING_FACILITY,
-    variants=[TutorialVariant()],
-)