fix : bug logur

Author: yassine abdou

grid2op/Backend/protectionScheme.py

@@ -1,5 +1,5 @@
 import copy
-from loguru import logger
+import logging
 from typing import Tuple, Union, Any, List, Optional
 import numpy as np
 
@@ -20,6 +20,7 @@ def __init__(
         parameters: Parameters,
         thermal_limits: ThermalLimits,
         is_dc: bool = False,
+        logger: Optional[logging.Logger] = None,
     ):
         """
         Initialise l'état du réseau avec des protections personnalisables.
@@ -42,6 +43,12 @@ def __init__(
         self._timestep_overflow = np.zeros(self.thermal_limits.n_line, dtype=dt_int)
         self.infos: List[str] = []
 
+        if logger is None:
+            self.logger = logging.getLogger(__name__)
+            self.logger.disabled = True
+        else:
+            self.logger: logging.Logger = logger.getChild("grid2op_BaseEnv")
+
     def _validate_input(self, backend: Backend, parameters: Optional[Parameters]) -> None:
         if not isinstance(backend, Backend):
             raise Grid2OpException(f"Argument 'backend' doit être de type 'Backend', reçu : {type(backend)}")
@@ -55,12 +62,18 @@ def _run_power_flow(self) -> Optional[Exception]:
         try:
             return self.backend._runpf_with_diverging_exception(self.is_dc)
         except Exception as e:
-            logger.error(f"Erreur flux de puissance : {e}")
+            if self.logger is not None:
+                self.logger.error(
+                    f"Erreur flux de puissance : {e}"
+                )
             return e
 
     def _update_overflows(self, lines_flows: np.ndarray) -> np.ndarray:
         if self._thermal_limit_a is None:
-            logger.error("Thermal limits must be provided for overflow calculations.")
+            if self.logger is not None:
+                self.logger.error(
+                    "Thermal limits must be provided for overflow calculations."
+                )
             raise ValueError("Thermal limits must be provided for overflow calculations.")
 
         lines_status = self.backend.get_line_status() # self._thermal_limit_a reste fixe. self._soft_overflow_threshold = 1 
@@ -77,7 +90,9 @@ def _disconnect_lines(self, lines_to_disconnect: np.ndarray, timestep: int) -> N
         for line_idx in np.where(lines_to_disconnect)[0]:
             self.backend._disconnect_line(line_idx)
             self.disconnected_during_cf[line_idx] = timestep
-            logger.warning(f"Ligne {line_idx} déconnectée au pas de temps {timestep}.")
+            if self.logger is not None:
+                self.logger.warning(f"Ligne {line_idx} déconnectée au pas de temps {timestep}.")         
+            
 
     def next_grid_state(self) -> Tuple[np.ndarray, List[Any], Union[None, Exception]]:
         try:
@@ -99,7 +114,8 @@ def next_grid_state(self) -> Tuple[np.ndarray, List[Any], Union[None, Exception]
             return self.disconnected_during_cf, self.infos, None
 
         except Exception as e:
-            logger.exception("Erreur inattendue dans le calcul de l'état du réseau.")
+            if self.logger is not None:
+                self.exception("Erreur inattendue dans le calcul de l'état du réseau.")
             return self.disconnected_during_cf, self.infos, e
 
 class NoProtection:

grid2op/Backend/thermalLimits.py

@@ -4,7 +4,7 @@
 except ImportError:
     from typing_extensions import Self
 
-from loguru import logger
+import logging
 import numpy as np
 import copy
 
@@ -21,7 +21,8 @@ def __init__(
             self,
             _thermal_limit_a: Optional[np.ndarray] = None,
             line_names: Optional[List[str]] = None,
-            n_line: Optional[int] = None
+            n_line: Optional[int] = None,
+            logger: Optional[logging.Logger] = None,
         ):
         """
         Initializes the thermal limits manager.
@@ -43,8 +44,12 @@ def __init__(
         self._n_line = n_line
         self._name_line = line_names
 
-        logger.info(f"ThermalLimits initialized with {self.n_line} limits.")
-
+        if logger is None:
+            self.logger = logging.getLogger(__name__)
+            self.logger.disabled = True
+        else:
+            self.logger: logging.Logger = logger.getChild("grid2op_BaseEnv")
+        
     @property
     def n_line(self) -> int:
         return self._n_line
@@ -54,8 +59,9 @@ def n_line(self, new_n_line: int) -> None:
         if new_n_line <= 0:
             raise ValueError("Number of lines must be a positive integer.")
         self._n_line = new_n_line
-        logger.info(f"Number of lines updated to {self._n_line}.")
-    
+        if self.logger is not None:
+            self.logger.info(f"Number of lines updated to {self._n_line}.")
+
     @property
     def name_line(self) -> Union[List[str], np.ndarray]:
         return self._name_line
@@ -75,7 +81,8 @@ def name_line(self, new_name_line: Union[List[str], np.ndarray]) -> None:
             raise ValueError("Length of name list must match the number of lines.")
 
         self._name_line = new_name_line
-        logger.info(f"Power line names updated")
+        if self.logger is not None:
+            self.logger.info("Power line names updated")
 
     @property
     def limits(self) -> np.ndarray:
@@ -185,7 +192,8 @@ def env_limits(self, thermal_limit):
             )
 
         self._thermal_limit_a = tmp
-        logger.info("Env thermal limits successfully set.")
+        if self.logger is not None:
+            self.logger.info("Env thermal limits successfully set.")
 
     def update_limits_from_vector(self, thermal_limit_a: np.ndarray) -> None:
         """
@@ -196,7 +204,8 @@ def update_limits_from_vector(self, thermal_limit_a: np.ndarray) -> None:
         """
         thermal_limit_a = np.array(thermal_limit_a).astype(dt_float)
         self._thermal_limit_a = thermal_limit_a
-        logger.info("Thermal limits updated from vector.")
+        if self.logger is not None:
+            self.logger.info("Thermal limits updated from vector.")
 
     def update_limits(self, env : "grid2op.Environment.BaseEnv") -> None:
         pass