fix logging

huggingface · Feb 19, 2024 · 52ea94d · 52ea94d
1 parent ada6525
commit 52ea94d
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 6 deletions.
diff --git a/optimum_benchmark/launchers/process/launcher.py b/optimum_benchmark/launchers/process/launcher.py
@@ -42,9 +42,6 @@ def launch(self, worker: Callable, *worker_args) -> BenchmarkReport:
             while not process_context.join():
                 pass
 
-        # restore the original logging configuration
-        setup_logging(log_level)
-
         report: BenchmarkReport = queue.get()
 
         return report

diff --git a/optimum_benchmark/launchers/torchrun/launcher.py b/optimum_benchmark/launchers/torchrun/launcher.py
@@ -57,9 +57,6 @@ def launch(self, worker: Callable, *worker_args) -> Dict[str, Any]:
                 entrypoint=entrypoint, args=(worker, queue, lock, log_level, *worker_args), config=launch_config
             )
 
-        # restore the original logging configuration
-        setup_logging(log_level)
-
         reports: List[BenchmarkReport] = []
         while not queue.empty():
             reports.append(queue.get())