Skip to content

Commit 18e7c8a

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent d0167dd commit 18e7c8a

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/core/test_loss.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)