Skip to content

Hf text test

datamodule #

datamodule(
    datamodule_config: str | None,
    overrides: list[str] | None,
) -> DataModule

Fixture that creates the datamodule for the given config.

test_dataset_location #

test_dataset_location(prepared_datamodule: HFDataModule)

Test that the dataset is downloaded to the correct location.

test_pretrained_weight_location #

test_pretrained_weight_location(
    prepared_datamodule: HFDataModule,
)

Test that the pretrained weights are downloaded to the correct location.