fix lint

Author: meichangsu1

src/twinkle/model/transformers/transformers.py

@@ -496,7 +496,7 @@ def calculate_loss(self, **kwargs):
         optimizer_config = self.optimizer_group[adapter_name]
         optimizer_config.train_status.num_tokens += counts.item()
         optimizer_config.train_status.loss_value += loss_value
-        optimizer_config.outputs['loss'] = optimizer_config.train_status.loss_value
+        outputs['loss'] = optimizer_config.train_status.loss_value
         return optimizer_config.train_status.loss_value.item()
 
     @remote_function()

tests/processor/test_processor.py

@@ -4,7 +4,6 @@
 import torch
 
 import twinkle
-from twinkle import DeviceMesh
 from twinkle.processor import GRPOLossProcessor, InputProcessor
 
 twinkle.initialize(mode='local')