Make a __init__.py file inside the directory so Python will treat that directory as a package
(that __init__.py can be empty)

P.S: run the code anyway if you still had any import error and look if it fixed or not