I was able to run the Linear flow example. One thing that bothered me was that there are no well-defined "input" and "output" data for the tasks I can visualize in the UI. For example.

@kubernetes(secrets="s3", memory=500, cpu=0.1)
    @step
    def start(self):
        self.my_var = "hello world"
        self.next(self.a)

    @kubernetes(secrets="s3", memory=500, cpu=0.1)
    @step
    def a(self):
        print("the data artifact is: %s" % self.my_var)
        self.next(self.end)

In the above snippet,

start

created

my_var

and it was used in

a

. I cannot find the visualization of this dataflow in the UI. Is this intended? I have added a reference screenshot from Flyte as an example.