def datamodule(dataset_path: Path) -> MVTecAD2:
        """Create and return a MVTec AD 2 datamodule."""
        _datamodule = MVTecAD2(
            root=dataset_path / "mvtecad2",
            category="dummy",
            train_batch_size=4,
            eval_batch_size=4,
            augmentations=Resize((256, 256)),
        )
        _datamodule.setup()

        return _datamodule
Create and return a MVTec AD 2 datamodule.
tests/unit/data/datamodule/image/test_mvtecad2.py