fix path

drisspg · Aug 29, 2024 · 83f754e · 83f754e
1 parent a901fed
commit 83f754e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/transformer_nuggets/utils/benchmark.py b/transformer_nuggets/utils/benchmark.py
@@ -281,7 +281,7 @@ def profiler(
  output_dir = path.parent
  output_dir.mkdir(parents=True, exist_ok=True)
 
- logger.info(f"💾 Trace file 📄 saved to: {bcolors.OKGREEN}{output_dir}{bcolors.ENDC}")
+ logger.info(f"💾 Trace file 📄 saved to: {bcolors.OKGREEN}{path}{bcolors.ENDC}")
 
  def trace_handler(prof) -> None:
  prof.export_chrome_trace(path.as_posix())