* update config

* re-blacken python code

* temporarily disable date and diff_py_file

* skip a format

* for gpt-gen

* fix reshard

* adapt assign and shape op

* add dist_assign & unittest

* add conditional block unittest

* rename unittest