remove print statements from test file

jbloomAus · Oct 9, 2024 · c0dc5bf · c0dc5bf
1 parent 99ace75
commit c0dc5bf
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/tests/unit/training/test_cache_activations_runner.py b/tests/unit/training/test_cache_activations_runner.py
@@ -104,7 +104,6 @@ def test_load_cached_activations():
 
     tokens_in_buffer = n_batches_in_buffer * store_batch_size * context_size
     total_training_tokens = n_buffers * tokens_in_buffer
-    print(f"Total Training Tokens: {total_training_tokens}")
 
     # better if we can look at the files
     cached_activations_fixture_path = os.path.join(
@@ -233,17 +232,12 @@ def test_cache_activations_runner_with_valid_start_end_pos_offset(tmp_path: Path
     start_pos_offset = 12
     end_pos_offset = 12
     training_context_size = context_size - start_pos_offset - end_pos_offset
-    print(f"n tokens per context: {context_size}")
     n_batches_in_buffer = 32
-    print(f"n batches in buffer: {n_batches_in_buffer}")
     store_batch_size = 1
-    print(f"store_batch_size: {store_batch_size}")
     n_buffers = 3
-    print(f"n_buffers: {n_buffers}")
 
     tokens_in_buffer = n_batches_in_buffer * store_batch_size * training_context_size
     total_training_tokens = n_buffers * tokens_in_buffer
-    print(f"Total Training Tokens: {total_training_tokens}")
 
     # better if we can look at the files (change tmp_path to a real path to look at the files)
     # tmp_path = os.path.join(os.path.dirname(__file__), "tmp")