Question on metaflow, while deploying it on AWS batch/step functions I do

python flow.py step-functions create

which packs the code and deploys it. But if I have some code referring to some other package like

x.yz.com.batch_local

here

from metaflow import FlowSpec,step
from x.yz.com.batch_local import batch


class TestFlow(FlowSpec):

    @batch()
    @step
    def start(self):
        self.next(self.doing_stuff)

    @batch()
    @step
    def doing_stuff(self):
        print('stuff')
        self.next(self.end)
    
    @batch()
    @step
    def end(self):
        print('end')
        pass

if __name__ == "__main__":
    TestFlow()

It doesn't detect the x package if I deploy the flow from the flow directory. Instead if I put the flow at the parent level of

x

folder, only then it recognizes x. Is this the right way to do it or am I missing something?