Move logger import

Author: yanxi-chen

trinity/algorithm/policy_loss_fn/gspo_policy_loss.py

@@ -9,7 +9,6 @@
 
 from trinity.algorithm.policy_loss_fn.policy_loss_fn import PolicyLossFn
 from trinity.algorithm.utils import aggregate_loss, masked_mean
-from trinity.utils.log import get_logger
 
 
 class GSPOLossFn(PolicyLossFn):
@@ -33,6 +32,8 @@ def __init__(
         self.clip_range_high = _clip_range_high
 
         if loss_agg_mode != "seq-mean-token-mean":
+            from trinity.utils.log import get_logger
+
             logger = get_logger(__name__)
             logger.warning(
                 f"The original GSPO paper requires loss_agg_mode to be 'seq-mean-token-mean', but the current setting is '{loss_agg_mode}'."