File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -452,8 +452,8 @@ def test_dataset_item_mismatched_lengths(self):
452452 eval_dataset_mask_shorter = [item_mask_shorter ]
453453 eval_dataset_ids_shorter = [item_ids_shorter ]
454454
455- self .mock_tokenizer .decode .side_effect = (
456- lambda ids , skip_special_tokens : "a" * len (ids )
455+ self .mock_tokenizer .decode .side_effect = lambda ids , skip_special_tokens : (
456+ "a" * len (ids )
457457 )
458458
459459 # Scenario 1: target_mask is shorter
@@ -468,8 +468,8 @@ def test_dataset_item_mismatched_lengths(self):
468468 self .mock_tokenizer .decode .assert_called_with ([2 , 3 ], skip_special_tokens = True )
469469
470470 self .mock_tokenizer .reset_mock () # Reset mock for the next scenario
471- self .mock_tokenizer .decode .side_effect = (
472- lambda ids , skip_special_tokens : "b" * len (ids )
471+ self .mock_tokenizer .decode .side_effect = lambda ids , skip_special_tokens : (
472+ "b" * len (ids )
473473 )
474474
475475 # Scenario 2: input_ids is shorter
You can’t perform that action at this time.
0 commit comments