indent error fix

torch/optim/lr_scheduler.py

@@ -198,8 +198,7 @@ def __init__(self, optimizer, T_max, eta_min=0, T_mult=2, restart=False, last_ep
         super(CosineAnnealingLR, self).__init__(optimizer, last_epoch)
 
     def get_lr(self):
-        new_lrs = [self.eta_min + (base_lr - self.eta_min) * (1 + math.cos(math.pi * self.last_epoch / self.T_max)) / 2 
-        for base_lr in self.base_lrs]
+        new_lrs = [self.eta_min + (base_lr - self.eta_min) * (1 + math.cos(math.pi * self.last_epoch / self.T_max)) / 2 for base_lr in self.base_lrs]
         if self.restart and self.last_epoch == self.T_max:
             self.last_epoch = 0
             self.T_max *= self.T_mult