Never struggle again to share data between your Kubeflow Pipelines components

produce_small_data_using_python_function.py

def get_model_location(bucket_name: str, model_folder: str) -> NamedTuple('returns', [('location', 'GCSPath')]):
    from google.cloud import storage

    folder = 'model/{}/export/exporter/'.format(model_folder)
    storage_client = storage.Client()
    bucket = storage_client.get_bucket(bucket_name)
    blobs = bucket.list_blobs(prefix=folder)
    blob_name = None
    for blob in blobs:
        tmp = len(blob.name.rstrip('/').split('/'))
        if tmp == 5:
            blob_name = blob.name
    print('Model location: {}'.format(blob_name))
    print('Model full location: gs://{0}/{1}'.format(bucket_name, blob_name))

    from collections import namedtuple
    output = namedtuple('returns', ['location'])
    return output('gs://{0}/{1}'.format(bucket_name, blob_name))