style: fix PLW1508

Author: kozistr

pytorch_optimizer/optimizer/lomo.py

@@ -41,7 +41,7 @@ def __init__(
         self.clip_grad_norm = clip_grad_norm
         self.clip_grad_value = clip_grad_value
 
-        self.local_rank: int = int(os.environ.get('LOCAL_RANK', 0))
+        self.local_rank: int = int(os.environ.get('LOCAL_RANK', '0'))
 
         self.gather_norm: bool = False
         self.grad_norms: List[torch.Tensor] = []
@@ -256,7 +256,7 @@ def __init__(
         self.grad_norms: List[torch.Tensor] = []
         self.clip_coef: Optional[float] = None
 
-        self.local_rank: int = int(os.environ.get('LOCAL_RANK', 0))
+        self.local_rank: int = int(os.environ.get('LOCAL_RANK', '0'))
         self.zero3_enabled: bool = is_deepspeed_zero3_enabled()
 
         self.grad_func: Callable[[Any], Any] = self.fuse_update_zero3() if self.zero3_enabled else self.fuse_update()

pytorch_optimizer/optimizer/muon.py

@@ -73,8 +73,8 @@ def __init__(
         adamw_params = self.get_parameters(adamw_params) if adamw_params is not None else []
         params.extend(adamw_params)
 
-        self.world_size: int = int(os.environ.get('WORLD_SIZE', 1))
-        self.rank: int = int(os.environ.get('RANK', 0))
+        self.world_size: int = int(os.environ.get('WORLD_SIZE', '1'))
+        self.rank: int = int(os.environ.get('RANK', '0'))
 
         defaults: DEFAULTS = {
             'lr': lr,