Fix: Enable Energy Calculation in Benchmarking by Implementing Subtra…

…ction Method (#149)

optimum_benchmark/trackers/energy.py

@@ -52,6 +52,19 @@ def log(self, prefix: str = "forward"):
         LOGGER.info(f"\t\t+ {prefix} RAM energy: {self.ram:f} ({self.unit})")
         LOGGER.info(f"\t\t+ {prefix} total energy: {self.total:f} ({self.unit})")
 
+    def __sub__(self, other: "Energy") -> "Energy":
+        """Enables subtraction of two Energy instances using the '-' operator."""
+        if self.unit != other.unit:
+            raise ValueError("Energy units must match to perform subtraction")
+
+        return Energy(
+            cpu=self.cpu - other.cpu,
+            gpu=self.gpu - other.gpu,
+            ram=self.ram - other.ram,
+            total=self.total - other.total,
+            unit=self.unit
+        )
+
 
 @dataclass
 class Efficiency: