from diffusers.modular_pipelines import SequentialPipelineBlocks
from .modular_blocks import ACTION2VIDEO_BLOCKS

class MatrixGameWanBlocks(SequentialPipelineBlocks):
    block_classes = list(ACTION2VIDEO_BLOCKS.copy().values())
    block_names = list(ACTION2VIDEO_BLOCKS.copy().keys())