ImportError: No module named enum

[ACFancy]

when I run the py:
bazel-bin/tensorflow/python/tools/print_selective_registration_header
--graphs=path/xx.pb > ops_to_register.h
there's the problem below:
ImportError: No module named enum

How to fix this problem, help!

[ACFancy]

fixed this change to python3 config, or change enum to aenum in Python2