Add transparent setting

Author: john-b-yang

codeclash/tournaments/pvp.py

@@ -17,7 +17,7 @@
 from codeclash.tournaments.tournament import AbstractTournament
 from codeclash.utils.atomic_write import atomic_write
 from codeclash.utils.aws import is_running_in_aws_batch, s3_log_sync
-from codeclash.utils.environment import copy_to_container
+from codeclash.utils.environment import copy_between_containers, copy_to_container
 
 
 class PvpTournament(AbstractTournament):
@@ -53,6 +53,10 @@ def metadata_file(self) -> Path:
     def rounds(self) -> int:
         return self.config["tournament"]["rounds"]
 
+    @property
+    def transparent(self) -> bool:
+        return self.config["tournament"].get("transparent", False)
+
     def get_metadata(self) -> dict:
         # will be saved in end()
         return {
@@ -120,6 +124,21 @@ def run_edit_phase(self, round_num: int) -> None:
             )
         self._compress_round_folder(round_num - 1)
 
+        if self.transparent:
+            # Copy agent's codebase to all other agents
+            self.logger.info("Transparent mode enabled: copying codebases between agents...")
+            for idx in range(len(self.agents)):
+                agent = self.agents[idx]
+                opponents = [a for j, a in enumerate(self.agents) if j != idx]
+                self.logger.info(f"Copying {agent.name}'s codebase to other agents...")
+                for opp in opponents:
+                    copy_between_containers(
+                        agent.environment,
+                        opp.environment,
+                        agent.environment.config.cwd,
+                        f"/{agent.name}/",
+                    )
+
         with ThreadPoolExecutor() as executor:
             futures = [executor.submit(self.run_agent, agent, round_num) for agent in self.agents]
             for future in futures:

codeclash/utils/environment.py

@@ -27,7 +27,9 @@ def copy_between_containers(
     Be extremely careful with trailing slashes in src_path and dest_path, the behavior
     of docker cp is also different depending on whether the destination exists.
     """
-    print(f"Copy between containers: {src_path} -> {dest_path}")
+    print(
+        f"Copy between containers: {src_container.container_id}:{src_path} -> {dest_container.container_id}:{dest_path}"
+    )
     # Some weird stuff happening on AWS where /tmp doesn't work properly
     dir = Path.home() / "tmp"
     dir.mkdir(parents=True, exist_ok=True)

main.py

@@ -36,6 +36,7 @@ def get_output_path() -> Path:
         else:
             timestamp = time.strftime("%y%m%d%H%M%S")
         rounds = config["tournament"]["rounds"]
+        transparent = config["tournament"].get("transparent", False)
         sims = config["game"]["sims_per_round"]
 
         players = [p["name"] for p in config["players"]]
@@ -45,6 +46,8 @@ def get_output_path() -> Path:
         folder_name = (
             f"PvpTournament.{config['game']['name']}.r{rounds}.s{sims}.p{p_num}.{p_list}{suffix_part}.{timestamp}"
         )
+        if transparent:
+            folder_name += ".transparent"
         if is_running_in_aws_batch():
             # Also add a UUID just to be safe
             _uuid = str(uuid.uuid4())