ma_policy/load_policy.py (1 line):
	- line 56: # TODO this will probably need to be changed when trying to run policy on GPU


ma_policy/util.py (1 line):
	- line 41: TODO: Revisit whether this is the right choice.