diff options
author | Guillaume Klein <guillaumekln@users.noreply.github.com> | 2022-03-16 18:03:55 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-03-16 18:03:55 +0300 |
commit | c2ee3e680f049c0d22e58d94d2161eb529aac3db (patch) | |
tree | ffdd7cc178152209fac8389be21dc184c7b9927e /tests | |
parent | 3e2147bcbf3d9897c42960fe21ab6116dc649897 (diff) |
Manage empty inputs in sample method (#677)
* Manage empty inputs in sample method
* Cleanup JobCreator methods
* Update test
* Also handle return_alternatives case in sample method
Diffstat (limited to 'tests')
-rw-r--r-- | tests/batching_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/batching_test.cc b/tests/batching_test.cc index f1f3d1f6..3cd6eb6b 100644 --- a/tests/batching_test.cc +++ b/tests/batching_test.cc @@ -23,7 +23,7 @@ TEST(BatchingTest, RebatchInput) { const std::vector<std::vector<size_t>> expected_batches = { {5, 4}, {1, 0}, - {2} + {2, 3} }; const auto batches = rebatch_input(load_examples({source, target}), 2, BatchType::Examples); |