Upload pkl predictions

Hi people!

I am trying to upload a prediction function using cloudpickle. It works fine if all the code is in the same file, but when creating a folder structure, the pkl is uploaded but it can not be extracted correctly as it miss modules. If the main file is like:

from my_module import my_func

def predict(live_features: pd.DataFrame) → pd.DataFrame:
features = my_func(live_features)
live_predictions = model.predict(lfeatures)
submission = pd.Series(live_predictions, index=live_features.index)
return submission.to_frame(“prediction”)

In this case how I should proceed? Numerai tells me “my_module” is not found

Thanks in advance

I think you have to keep all the code in the same file. Maybe someone can prove me wrong?