# Copyright (c) 2022 Zhipu.AI import sys if sys.argv[1] == 'block': from test.test_block import main main() elif sys.argv[1] == 'rel_shift': from test.test_rel_shift import main main()