Fix: Remove train_iter from grain processing test

bvandermoon · bvandermoon · commit c9d0b113774a · 2026-05-13T21:42:35.000Z
diff --git a/tests/unit/grain_data_processing_test.py b/tests/unit/grain_data_processing_test.py
@@ -464,7 +464,6 @@ def setUp(self):
         tokenizer_path=os.path.join(MAXTEXT_ASSETS_ROOT, "tokenizers", "tokenizer.default"),
         enable_checkpointing=False,
     )
-    self.train_iter = grain_data_processing.make_grain_train_iterator(self.config, self.mesh, self.process_indices)
 
 
 @pytest.mark.external_training

Original file line number	Diff line number	Diff line change
`@@ -464,7 +464,6 @@ def setUp(self):`
`464`	`464`	`tokenizer_path=os.path.join(MAXTEXT_ASSETS_ROOT, "tokenizers", "tokenizer.default"),`
`465`	`465`	`enable_checkpointing=False,`
`466`	`466`	`)`
`467`		`- self.train_iter = grain_data_processing.make_grain_train_iterator(self.config, self.mesh, self.process_indices)`
`468`	`467`
`469`	`468`
`470`	`469`	`@pytest.mark.external_training`