mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
9b736c707c
Differential Revision: D51149464 Pull Request resolved: https://github.com/pytorch/pytorch/pull/113357 Approved by: https://github.com/huydhn
29 lines
751 B
Python
29 lines
751 B
Python
import argparse
|
|
import os
|
|
import os.path
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--input-file")
|
|
parser.add_argument("--output-file")
|
|
parser.add_argument("--install-dir", "--install_dir")
|
|
parser.add_argument("--replace", action="append", nargs=2)
|
|
options = parser.parse_args()
|
|
|
|
with open(options.input_file) as f:
|
|
contents = f.read()
|
|
|
|
output_file = os.path.join(options.install_dir, options.output_file)
|
|
os.makedirs(os.path.dirname(output_file), exist_ok=True)
|
|
|
|
for old, new in options.replace:
|
|
contents = contents.replace(old, new)
|
|
|
|
with open(output_file, "w") as f:
|
|
f.write(contents)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() # pragma: no cover
|