Linear metaflow transition.

metaflow_1.py

import numpy as np

from metaflow import FlowSpec, step


class CalculateMean(FlowSpec):
    @step
    def start(self):
      """
      Initializes a random dataset.
      """
      np.random.seed(0)
      self.data = np.random.rand(100)
      self.next(self.mean)

    @step
    def mean(self):
      """
      Calculates the mean.
      """
      self.avg = np.mean(self.data)
      self.next(self.end)

    @step
    def end(self):
      """
      Prints the results and terminates the run.
      """
      print(f'Mean = {self.avg}')


if __name__ == '__main__':
    CalculateMean()