Hf example test TestHFExample # Bases: LearningAlgorithmTests[HFExample] Tests for the HF example. test_overfit_batch # test_overfit_batch( algorithm: HFExample, datamodule: HFDataModule, accelerator: str, devices: int | list[int], training_batch: dict[str, Tensor], num_steps: int = 3, ) Test that the loss decreases on a single batch.