Consider TensorConstant for the workaround too.
some pep8