# Import the datamodule
from anomalib.data import Folder
from anomalib.data.utils import TestSplitMode, ValSplitMode

# Create the datamodule
datamodule = Folder(
    name="hazelnut_toy",
    root="datasets/hazelnut_toy",
    normal_dir="good",
    val_split_mode=ValSplitMode.NONE,
    test_split_mode=TestSplitMode.NONE,
    task="classification",
)

# Setup the datamodule
datamodule.setup()
