from metaflow import FlowSpec, step, card, current, conda_base
from metaflow.user_decorators.user_step_decorator import StepMutator


class AddTestCard(StepMutator):
    def mutate(self, mutable_step):
        from metaflow import card as card_deco

        mutable_step.add_decorator(card_deco, deco_kwargs={"id": "test_card"})

@conda_base
class ReproFlow(FlowSpec):
    @card
    @AddTestCard()
    @step
    def start(self):
        print(type(current.card["test_card"]))
        self.next(self.end)

    @step
    def end(self):
        pass


if __name__ == "__main__":
    ReproFlow()