test

huggingface · Oct 3, 2024 · 0a334f2 · 0a334f2
1 parent e1b5c68
commit 0a334f2
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/optimum_benchmark/benchmark/report.py b/optimum_benchmark/benchmark/report.py
@@ -50,13 +50,11 @@ def aggregate(measurements: List["Measurements"]) -> "Measurements":
 
     def to_plain_text(self) -> str:
         plain_text = ""
-        i = 0
 
         for key in ["memory", "latency", "throughput", "energy", "efficiency"]:
             measurement = getattr(self, key)
             if measurement is not None:
-                i += 1
-                plain_text += f"\t+ {i}. {key}:\n"
+                plain_text += f"\t+ {key}:\n"
                 plain_text += measurement.to_plain_text()
 
         return plain_text
@@ -68,13 +66,11 @@ def log(self):
 
     def to_markdown_text(self) -> str:
         markdown_text = ""
-        i = 0
 
         for key in ["memory", "latency", "throughput", "energy", "efficiency"]:
             measurement = getattr(self, key)
             if measurement is not None:
-                i += 1
-                markdown_text += f"{i}. {key}:\n\n"
+                markdown_text += f"## {key}:\n\n"
                 markdown_text += measurement.to_markdown_text()
 
         return markdown_text