File size: 983 Bytes
685ce0f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
#!/usr/bin/env python3
import sys
name = sys.argv[1]
processor_class = sys.argv[2]
model_class = sys.argv[3]
checkpoint = sys.argv[4]
mask = sys.argv[5]
with open("/home/patrick/transformers/src/transformers/file_utils.py", "r") as f:
lines = f.readlines()
format_dict = {"processor_class": processor_class, "model_class": model_class, "checkpoint": checkpoint, "mask": mask}
with open("./example.py", "w") as f:
f.write("#!/usr/bin/env python3\n")
is_in = False
is_in_code = False
for line in lines:
if line.strip() == (name + ' = r"""'):
is_in = True
if is_in and "```python" in line:
is_in_code = True
if is_in_code:
if ">>>" in line:
f.write(line.split(">>> ")[-1].format(**format_dict))
elif "..." in line:
f.write(line.split("... ")[-1].format(**format_dict))
if is_in_code and (line.strip() == '"""'):
is_in = is_in_code = False
|