fix test

stanford-crfm · Oct 11, 2024 · bbc3d56 · bbc3d56
1 parent 2b7712d
commit bbc3d56
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_new_cache.py b/tests/test_new_cache.py
@@ -505,6 +505,9 @@ def test_sharded_cache_writer_trims_on_resume():
             for ex in batched(source.open_shard(shard_name), 8):
                 writer.write_batch(shard_name, processor(ex))
 
+        for shard_name in source.shard_names:
+            writer.finish_shard(shard_name, 10)
+
         writer.finish()
 
         # now deliberately truncate the ledger a bit