text: Title: error when running training/main.py
  Body: I am getting a ModuleNotFoundError for training when running src/training/main.py. It points to line 19 in main.py, the import function. 
  
  Edit: Fixed it. Forgot to add pythonpath