Allow simulating idle components

Author: PedramBakh

carbontracker/components/component.py

@@ -50,13 +50,13 @@ def handlers_by_name(
 ):
     for comp in COMPONENTS:
         if comp["name"] == name:
-            if name == "cpu" and sim_cpu and sim_cpu_tdp:
+            if name == "cpu" and sim_cpu is not None and sim_cpu_tdp is not None:
                 return [lambda pids, devices_by_pid: SimulatedCPUHandler(
                     sim_cpu, 
                     float(sim_cpu_tdp),
                     float(sim_cpu_util) if sim_cpu_util is not None else 0.5
                 )]
-            elif name == "gpu" and sim_gpu and sim_gpu_watts:
+            elif name == "gpu" and sim_gpu is not None and sim_gpu_watts is not None:
                 return [lambda pids, devices_by_pid: SimulatedGPUHandler(
                     sim_gpu, 
                     float(sim_gpu_watts),

carbontracker/components/cpu/sim_cpu.py

@@ -6,8 +6,8 @@ def __init__(self, name: str, tdp: float, utilization: float = 0.5):
         super().__init__(pids=[], devices_by_pid=False)
         if not isinstance(name, str) or not name.strip():
             raise ValueError("CPU name must be a non-empty string.")
-        if tdp is None or not isinstance(tdp, (int, float)) or tdp <= 0:
-            raise ValueError("CPU TDP must be a positive number.")
+        if tdp is None or not isinstance(tdp, (int, float)) or tdp < 0:
+            raise ValueError("CPU TDP must be a non-negative number.")
         if not isinstance(utilization, (int, float)) or not (0.0 <= utilization <= 1.0):
             raise ValueError("CPU utilization must be between 0.0 and 1.0.")
         self.cpu_brand = name

carbontracker/components/gpu/sim_gpu.py

@@ -6,8 +6,8 @@ def __init__(self, name: str, watts: float, utilization: float = 0.5):
         super().__init__(pids=[], devices_by_pid=False)
         if not isinstance(name, str) or not name.strip():
             raise ValueError("GPU name must be a non-empty string.")
-        if watts is None or not isinstance(watts, (int, float)) or watts <= 0:
-            raise ValueError("GPU watts must be a positive number.")
+        if watts is None or not isinstance(watts, (int, float)) or watts < 0:
+            raise ValueError("GPU watts must be a non-negative number.")
         if not isinstance(utilization, (int, float)) or not (0.0 <= utilization <= 1.0):
             raise ValueError("GPU utilization must be between 0.0 and 1.0.")
         self.gpu_brand = name
@@ -26,5 +26,20 @@ def power_usage(self) -> List[float]:
     def init(self):
         print(f"Using simulated GPU: {self.gpu_brand} with power consumption: {self.watts:.2f}W (at {self.utilization*100:.0f}% utilization)")
 
+    def shutdown(self):
+        pass
+
+    def devices(self) -> List[str]:
+        return [self.gpu_brand]
+
+    def available(self) -> bool:
+        return True
+
+    def power_usage(self) -> List[float]:
+        return [self.watts]
+
+    def init(self):
+        print(f"Using simulated GPU: {self.gpu_brand} with power consumption: {self.watts:.2f}W (at {self.utilization*100:.0f}% utilization)")
+
     def shutdown(self):
         pass