workflow.py

class Workflow:
    def __init__(self, name: str, *stages: Stage) -> None:
        self.name = name
        if not stages:
            raise ValueError('Stages must be supplied')
        self.stages = stages


class Stage:
    def __init__(self, name: str) -> None:
        self.name = name