source/inference.py (3 lines):
	- line 34: # from sagemaker.content_types import CONTENT_TYPE_JSON, CONTENT_TYPE_CSV, CONTENT_TYPE_NPY # TODO: for local debug only. Remove or comment when deploying remotely.
	- line 35: # from six import StringIO, BytesIO  # TODO: for local debug only. Remove or comment when deploying remotely.
	- line 44: # TODO: Maybe put this from SM env vars


source/train.py (2 lines):
	- line 309: #         TODO: Do we want to augment the test split too?
	- line 352: parser.add_argument('--resume', type=str, default="True") # TODO: is it relevant?