Added more static types

Corrected usage of int compared to [int]

Author: root

codecarbon/emissions_tracker.py

@@ -176,7 +176,7 @@ def __init__(
             str
         ] = _sentinel,  # Deprecated, use electricitymaps_api_token
         tracking_mode: Optional[str] = _sentinel,
-        tracking_pids: Optional[int] = _sentinel,
+        tracking_pids: Optional[List[int]] = _sentinel,
         log_level: Optional[Union[int, str]] = _sentinel,
         on_csv_write: Optional[str] = _sentinel,
         logger_preamble: Optional[str] = _sentinel,
@@ -1182,7 +1182,7 @@ def track_emissions(
         str
     ] = _sentinel,  # Deprecated, use electricitymaps_api_token
     tracking_mode: Optional[str] = _sentinel,
-    tracking_pids: Optional[int] = _sentinel,
+    tracking_pids: Optional[List[int]] = _sentinel,
     log_level: Optional[Union[int, str]] = _sentinel,
     on_csv_write: Optional[str] = _sentinel,
     logger_preamble: Optional[str] = _sentinel,

codecarbon/external/ram.py

@@ -50,7 +50,7 @@ def __init__(
                                       total RAM used. Defaults to True.
             tracking_mode (str, optional): Whether to track "machine" or "process" RAM.
                                           Defaults to "machine".
-            tracking_pids (int, optional): Process id to track RAM usage for "process"
+            tracking_pids ([int], optional): Process id to track RAM usage for "process"
                                             tracking_mode. Defaults to None.
             force_ram_power (int, optional): User-provided RAM power in watts. If provided,
                                            this value is used instead of estimating RAM power.

tests/test_pid_tracking.py

@@ -57,7 +57,7 @@ def test_carbon_pid_tracking_offline(self):
             output_dir=self.emissions_path,
             output_file=self.emissions_file + "_global.csv",
             tracking_mode="process",
-            tracking_pids=os.getpid(),
+            tracking_pids=[os.getpid()],
         )
 
         tracker_pid.start()
@@ -76,4 +76,10 @@ def test_carbon_pid_tracking_offline(self):
         assert isinstance(emissions_pid, float)
 
         self.assertNotEqual(emissions_pid, 0.0)
-        self.assertAlmostEqual(emissions_self, emissions_pid, 2)
+
+        # Compare emissions from both trackers, should be less than 10% difference
+        diff = abs(emissions_pid - emissions_self)
+        avg = (emissions_pid + emissions_self) / 2
+        percent_diff = (diff / avg) * 100
+        print(f"Percent difference: {percent_diff}%")
+        self.assertLessEqual(percent_diff, 10.0)