Train Valid Test Python Version Update