test: add regression test for no_split_module_classes accepting set type (#4048)
* test: add regression test for no_split_module_classes accepting set type
get_balanced_memory and infer_auto_device_map should accept a set for
no_split_module_classes without raising 'TypeError: unhashable type: set'.
Bug was present in <=1.5.0, fixed in #2345, this test prevents regression.
* Apply style fixes
---------
Co-authored-by: 你的GitHub名稱 <你的GitHub信箱>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>