Adding back import that was accidentally removed.

python/unitytrainers/trainer_controller.py

 import logging
 import os
 import yaml
+import re
 
 import numpy as np
 import tensorflow as tf