from metaflow import (
    FlowSpec,
    step,
    batch,
)

custom_step_tags = {
    "goodbye": "world",
    "hello": "universe",
    }

class BatchTags2(FlowSpec):
    @step
    def start(self):
        self.next(self.hello)

    @batch(aws_batch_tags=custom_step_tags)
    @step
    def hello(self):
        self.next(self.end)

    @step
    def end(self):
        pass

if __name__ == "__main__":
    BatchTags2()