Have an issue with the @pypi decorator in Metaflow. Here is some simple code to demonstrate the issue:

from metaflow import FlowSpec, step, pypi, pypi_base

@pypi_base(python='3.12.3')

class dummy_test_pypi(FlowSpec):

@pypi(packages={'keras-cv': '0.9.0'})

@step

def start(self):

import keras_cv

print(keras_cv.__version__)

self.next(self.end)

@step

def end(self):

print("All done. \n\n Congratulations!\n")

return

if __name__ == '__main__':

dummy_test_pypi()