Merge pull request #3899 from AI-Hypercomputer:bvandermoon-test-fixes

Google-ML-Automation · Google-ML-Automation · commit 1ea03b1c2f90 · 2026-05-13T16:02:59.000-07:00
PiperOrigin-RevId: 915107454
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`