fixed multidomain data bug in predict.py