exposing verbose options

PiperOrigin-RevId: 799570510
Change-Id: Ia60b44ba993ab5347fc68fa01a294d84dba796bd

Author: DavidePaglieri

concordia/environment/engines/parallel.py

@@ -150,7 +150,10 @@ def terminate(
     return should_terminate_string == entity_lib.BINARY_OPTIONS['affirmative']
 
   def make_observation(
-      self, game_master: entity_lib.Entity, entity: entity_lib.Entity
+      self,
+      game_master: entity_lib.Entity,
+      entity: entity_lib.Entity,
+      verbose: bool = False,
   ) -> str:
     """Make an observation for a game object."""
     observation = game_master.act(
@@ -161,7 +164,12 @@ def make_observation(
             output_type=entity_lib.OutputType.MAKE_OBSERVATION,
         )
     )
-    print(f'Observation: {observation} for {entity.name}')
+    if verbose:
+      print(
+          termcolor.colored(
+              f'Observation: {observation} for {entity.name}', _PRINT_COLOR
+          )
+      )
     return observation
 
   @override
@@ -200,7 +208,7 @@ def run_loop(
       tasks = {}
       for entity in entities:
         tasks[entity.name] = functools.partial(
-            entity.observe, self.make_observation(game_master, entity)
+            entity.observe, self.make_observation(game_master, entity, verbose)
         )
       concurrency.run_tasks(tasks, executor=executor)
 

concordia/prefabs/simulation/questionnaire_simulation.py

@@ -48,6 +48,7 @@ def __init__(
       embedder: Callable[[str], np.ndarray],
       engine: parallel.ParallelQuestionnaireEngine | None = None,
       max_workers: int | None = None,
+      verbose: bool = False,
   ):
     """Initialize the simulation object.
 
@@ -68,10 +69,12 @@ def __init__(
         parallel.ParallelQuestionnaireEngine.
       max_workers: the maximum number of workers to use in the engine's
         ThreadPoolExecutor, if the default engine is used.
+      verbose: Whether to print verbose output.
     """
     self._config = config
     self._model = model
     self._embedder = embedder
+    self._verbose = verbose
     if engine is None:
       if not max_workers:
         self._engine = parallel.ParallelQuestionnaireEngine()
@@ -193,11 +196,13 @@ def add_entity(
     # Check if a pre-loaded memory state was passed in the entity's params.
     memory_state = instance_config.params.get("memory_state")
     if memory_state:
-      print(f"Found pre-loaded memory state for {entity.name}. Setting it.")
+      if self._verbose:
+        print(f"Found pre-loaded memory state for {entity.name}. Setting it.")
       try:
         memory_component = entity.get_component("__memory__")
         memory_component.set_state(memory_state)
-        print(f"Successfully set pre-loaded memories for {entity.name}.")
+        if self._verbose:
+          print(f"Successfully set pre-loaded memories for {entity.name}.")
       except (KeyError, TypeError, ValueError) as e:
         print(f"Error setting pre-loaded memory for {entity.name}: {e}")
 
@@ -372,7 +377,8 @@ def save_checkpoint(self, step: int, checkpoint_path: str):
     try:
       with open(checkpoint_file, "w") as f:
         json.dump(checkpoint_data, f, indent=2)
-      print(f"Step {step}: Saved checkpoint to {checkpoint_file}")
+      if self._verbose:
+        print(f"Step {step}: Saved checkpoint to {checkpoint_file}")
     except IOError as e:
       print(f"Error saving checkpoint at step {step}: {e}")