Update diag.py

from __future__ import print_function