AICup/code/__init__.py

import os

if 'AI' in os.getcwd():
    from code.TSP_solver import *
    from code.io_tsp import *
else:
    from AI2019.code.TSP_solver import *
    from AI2019.code.io_tsp import *