add clean mode for experiment to avoid printing any NeMo-Run specific logs (#208)

Signed-off-by: Hemil Desai <[email protected]>

Author: hemildesai

nemo_run/run/experiment.py

@@ -30,6 +30,7 @@
 
 import fiddle as fdl
 import networkx as nx
+import rich
 from fiddle._src import daglish, diffing
 from rich.console import Group
 from rich.live import Live
@@ -68,6 +69,18 @@
 )
 
 
+class DummyConsole:
+    """A dummy console that mimics rich.console.Console but does nothing."""
+
+    def __getattr__(self, name):
+        """Return a no-op function for any attribute access."""
+
+        def no_op(*args, **kwargs):
+            pass
+
+        return no_op
+
+
 class Experiment(ConfigurableMixin):
     """
     A context manager to launch and manage multiple runs, all using pure Python.
@@ -280,6 +293,7 @@ def __init__(
         _reconstruct: bool = False,
         jobs: list[Job | JobGroup] | None = None,
         base_dir: str | None = None,
+        clean_mode: bool = False,
     ) -> None:
         """
         Initializes an experiment run by creating its metadata directory and saving the experiment config.
@@ -294,6 +308,7 @@ def __init__(
             log_level: Set log level for the experiment. Defaults to WARN.
             _reconstruct: Generally, the user does not need to specify this flag.
                 This is only set to True when using run.Experiment.from_dir.
+            clean_mode: If True, disables all console output (logs, progress bars, etc.). Defaults to False.
         """
         configure_logging(level=log_level)
         self._reconstruct = _reconstruct
@@ -318,6 +333,9 @@ def __init__(
         self._jobs: list[Job | JobGroup] = jobs or []
         self.tunnels: dict[str, Tunnel] = {}
         self.console = CONSOLE
+        self.clean_mode = clean_mode
+        if self.clean_mode:
+            self.console = DummyConsole()
         self._launched = False
         self._live_progress = None
         self._current_experiment_token = None
@@ -329,6 +347,7 @@ def to_config(self) -> Config:
             id=self._id,
             executor=self.executor.to_config(),
             log_level=self.log_level,
+            clean_mode=self.clean_mode,
         )
 
     def _save_experiment(self, exist_ok: bool = False):
@@ -1037,9 +1056,10 @@ def _initialize_live_progress(self):
         if not self._live_progress:
             # Disable live progress if we are tailing logs for any task
             # as tty output consistency can not be guaranteed as of now
-            if any(map(lambda job: job.tail_logs, self.jobs)):
+            if self.clean_mode or any(map(lambda job: job.tail_logs, self.jobs)):
                 return
 
+            assert isinstance(self.console, rich.console.Console)
             self._progress = Progress(
                 "{task.description}",
                 SpinnerColumn(),