fix relative imports

[vratins]

currently I use hacky sys path edits to make sure imports don't fail, fixed it for dataset.py; need to do it throughout the codebase