Skip to content

Text classification test

datamodule #

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

Fixture that creates the datamodule for the given config.

test_dataset_location #

test_dataset_location(
    prepared_datamodule: TextClassificationDataModule,
)

Test that the dataset is downloaded to the correct location.

test_pretrained_weight_location #

test_pretrained_weight_location(
    prepared_datamodule: TextClassificationDataModule,
)

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