Custom modules?

[pr1ntr]

How do i package a prediction with my own python code?

i have structure like this

• 📂 root
  • 📂 weights
  • 📂 scripts
    • 📄 download_weights
  • 📄 predict.py
  • 📂 module
    • 📄 main.py

predict.py has something like

from module import main

errors when i try to run something from the container

File "/src/scripts/download_weights", line 5, in <module>
    from module import main

can someone lead me to some docs on how to deal with this?