Skip to content

Commit

Permalink
added comment
Browse files Browse the repository at this point in the history
  • Loading branch information
twerkmeister committed Sep 20, 2023
1 parent 8fc686f commit a54232c
Showing 1 changed file with 13 additions and 1 deletion.
14 changes: 13 additions & 1 deletion tests/nlu/classifiers/test_diet_classifier.py
Original file line number Diff line number Diff line change
Expand Up @@ -972,7 +972,19 @@ async def test_no_bilou_when_entity_recognition_off(
@pytest.mark.timeout(120, func_only=True)
@pytest.mark.parametrize(
"batch_size, expected_num_batches",
[(8, 6), (15, 3), (16, 3), (18, 3), (20, 2), (32, 1), (64, 1), (128, 1), (256, 1)],
# the training dataset has 48 NLU examples
[
(1, 48),
(8, 6),
(15, 3),
(16, 3),
(18, 3),
(20, 2),
(32, 1),
(64, 1),
(128, 1),
(256, 1),
],
)
async def test_dropping_of_last_partial_batch(
batch_size: int,
Expand Down

0 comments on commit a54232c

Please sign in to comment.