def segmentation_datamodule(folder_dataset_path: Path) -> AnomalibDataModule:
    """Create a segmentation datamodule for testing.

    The datamodule is created with a folder dataset, that has a mask directory.
    """
    # create the folder datamodule
    return Folder(
        name="seg_dataset",
        root=folder_dataset_path,
        normal_dir="good",
        abnormal_dir="bad",
        mask_dir="masks",  # include masks for segmentation dataset
        train_batch_size=1,
        eval_batch_size=1,
        num_workers=0,
    )
Create a segmentation datamodule for testing.

    The datamodule is created with a folder dataset, that has a mask directory.
    
tests/integration/test_task_types.py