Lint

Author: benoit-cty

codecarbon/core/cpu.py

@@ -16,6 +16,7 @@
 
 try:
     import psutil
+
     PSUTIL_AVAILABLE = True
 except ImportError:
     PSUTIL_AVAILABLE = False

codecarbon/core/resource_tracker.py

@@ -45,10 +45,12 @@ def set_CPU_tracking(self):
             max_power = self.tracker._force_cpu_power
         else:
             max_power = tdp.tdp * cpu_number if tdp.tdp is not None else None
-        
+
         # Check for forced constant mode first
         if self.tracker._conf.get("force_mode_constant", False):
-            logger.info("Force constant mode requested - bypassing psutil and using constant CPU power")
+            logger.info(
+                "Force constant mode requested - bypassing psutil and using constant CPU power"
+            )
             model = tdp.model
             if max_power is None and self.tracker._force_cpu_power:
                 max_power = self.tracker._force_cpu_power
@@ -63,7 +65,7 @@ def set_CPU_tracking(self):
             )
             self.tracker._hardware.append(hardware_cpu)
             return
-            
+
         if self.tracker._conf.get("force_mode_cpu_load", False) and (
             tdp.tdp is not None or self.tracker._force_cpu_power is not None
         ):

codecarbon/core/util.py

@@ -11,6 +11,7 @@
 
 try:
     import psutil
+
     PSUTIL_AVAILABLE = True
 except ImportError:
     PSUTIL_AVAILABLE = False
@@ -129,7 +130,7 @@ def count_cpus() -> int:
         # Fallback to using os.cpu_count() or physical CPU count
         cpu_count = os.cpu_count()
         return cpu_count if cpu_count is not None else 1
-    
+
     if SLURM_JOB_ID is None:
         return psutil.cpu_count()