mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-12-25 13:32:44 +00:00
7080a8edb0
* [workflows]bug_repair * [workflow]new_pr_fixing_bugs Co-authored-by: binmakeswell <binmakeswell@gmail.com>
20 lines
626 B
Python
20 lines
626 B
Python
import argparse
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--fileNameList', type=str)
|
|
args = parser.parse_args()
|
|
name_list = args.fileNameList.split(":")
|
|
folder_need_check = set()
|
|
for loc in name_list:
|
|
# Find only the sub-folder of 'example' folder
|
|
if loc.split("/")[0] == "examples" and len(loc.split("/")) >= 4:
|
|
folder_need_check.add(loc.split("/")[1] + "/" + loc.split("/")[2])
|
|
# Output the result using print. Then the shell can get the values.
|
|
print(list(folder_need_check))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|