|
|
|
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 |
|
|