Minor lint fix

odelalleau · odelalleau · commit 43f7eaecc5c1 · 2025-07-29T15:15:07.000-04:00
diff --git a/nemo_rl/algorithms/loss_functions.py b/nemo_rl/algorithms/loss_functions.py
@@ -624,7 +624,7 @@ def _dpo_loss(
         )
 
     # TODO a cleaner typing fix would be required (probably that DPOLossFn should not inherit from PreferenceLoss)
-    def __call__( # type: ignore
+    def __call__(  # type: ignore
         self,
         next_token_logits: Tensor,
         data: BatchedDataDict[DPOLossDataDict],

Original file line number	Diff line number	Diff line change
`@@ -624,7 +624,7 @@ def _dpo_loss(`
`624`	`624`	`)`
`625`	`625`
`626`	`626`	`# TODO a cleaner typing fix would be required (probably that DPOLossFn should not inherit from PreferenceLoss)`
`627`		`- def __call__( # type: ignore`
	`627`	`+ def __call__( # type: ignore`
`628`	`628`	`self,`
`629`	`629`	`next_token_logits: Tensor,`
`630`	`630`	`data: BatchedDataDict[DPOLossDataDict],`